File tree 4 files changed +387
-406
lines changed
4 files changed +387
-406
lines changed Original file line number Diff line number Diff line change @@ -26,6 +26,17 @@ set(CMAKE_CXX_STANDARD 20)
26
26
set (CMAKE_CXX_STANDARD_REQUIRED ON )
27
27
set (CMAKE_CXX_EXTENSIONS OFF )
28
28
29
+ # Make sure the folder is the same as the one CXX_COMPILER is in if ROCM_PATH is not set
30
+ if (NOT DEFINED ROCM_PATH)
31
+ get_filename_component (ROCM_PATH ${CMAKE_CXX_COMPILER} DIRECTORY )
32
+ get_filename_component (ROCM_PATH ${ROCM_PATH} ABSOLUTE )
33
+ string (REGEX MATCH ".*/rocm[^/]*/" ROCM_PREFIX ${ROCM_PATH} )
34
+ string (REGEX REPLACE "/$" "" ROCM_PATH ${ROCM_PREFIX} )
35
+ message (STATUS "ROCM_PATH not defined, inferred from CXX_COMPILER: ${ROCM_PATH} " )
36
+ endif ()
37
+
38
+ find_package (amd_comgr REQUIRED CONFIG HINTS ${ROCM_PATH} /lib/cmake/amd_comgr)
39
+
29
40
if (DEFINED Python_EXECUTABLE AND Python_EXECUTABLE)
30
41
message (STATUS "Manually set Python_EXECUTABLE to ${Python_EXECUTABLE} " )
31
42
endif ()
@@ -73,5 +84,7 @@ nanobind_add_module(rocisa NOMINSIZE NB_SUPPRESS_WARNINGS
73
84
${CMAKE_CURRENT_SOURCE_DIR} /rocisa/src/macro.cpp
74
85
${ROCISAINST_SOURCE}
75
86
${ROCISAPASS_SOURCE} )
87
+
76
88
target_include_directories (rocisa PRIVATE ${CMAKE_CURRENT_SOURCE_DIR} /rocisa/include )
89
+ target_link_libraries (rocisa PRIVATE amd_comgr)
77
90
set_target_properties (rocisa PROPERTIES LIBRARY_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR} /lib)
You can’t perform that action at this time.
0 commit comments