File: FindTPLROCM.cmake

package info (click to toggle)
kokkos 5.0.2-1
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 15,148 kB
  • sloc: cpp: 225,388; sh: 1,250; python: 78; makefile: 16; fortran: 4; ansic: 2
file content (16 lines) | stat: -rw-r--r-- 447 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
 
include(FindPackageHandleStandardArgs)

find_library(AMD_HIP_LIBRARY amdhip64 PATHS ENV ROCM_PATH PATH_SUFFIXES lib)
find_library(HSA_RUNTIME_LIBRARY hsa-runtime64 PATHS ENV ROCM_PATH PATH_SUFFIXES lib)

find_package_handle_standard_args(TPLROCM DEFAULT_MSG AMD_HIP_LIBRARY HSA_RUNTIME_LIBRARY)

kokkos_create_imported_tpl(
  ROCM
  INTERFACE
  LINK_LIBRARIES
  ${HSA_RUNTIME_LIBRARY}
  ${AMD_HIP_LIBRARY}
  COMPILE_DEFINITIONS
  __HIP_ROCclr__
)