Skip to content

Commit 20eb68a

Browse files
committed
[rocisa] Use comgr instead of calling amdclang++
1 parent 204d6d3 commit 20eb68a

File tree

4 files changed

+387
-406
lines changed

4 files changed

+387
-406
lines changed

tensilelite/rocisa/CMakeLists.txt

+13
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,17 @@ set(CMAKE_CXX_STANDARD 20)
2626
set(CMAKE_CXX_STANDARD_REQUIRED ON)
2727
set(CMAKE_CXX_EXTENSIONS OFF)
2828

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+
2940
if(DEFINED Python_EXECUTABLE AND Python_EXECUTABLE)
3041
message(STATUS "Manually set Python_EXECUTABLE to ${Python_EXECUTABLE}")
3142
endif()
@@ -73,5 +84,7 @@ nanobind_add_module(rocisa NOMINSIZE NB_SUPPRESS_WARNINGS
7384
${CMAKE_CURRENT_SOURCE_DIR}/rocisa/src/macro.cpp
7485
${ROCISAINST_SOURCE}
7586
${ROCISAPASS_SOURCE})
87+
7688
target_include_directories(rocisa PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/rocisa/include)
89+
target_link_libraries(rocisa PRIVATE amd_comgr)
7790
set_target_properties(rocisa PROPERTIES LIBRARY_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/lib)

0 commit comments

Comments
 (0)