diff --git a/projects/aqlprofile/src/core/tests/CMakeLists.txt b/projects/aqlprofile/src/core/tests/CMakeLists.txt index a1ae35ecff..0e0e40831f 100644 --- a/projects/aqlprofile/src/core/tests/CMakeLists.txt +++ b/projects/aqlprofile/src/core/tests/CMakeLists.txt @@ -4,14 +4,15 @@ include(GoogleTest) find_package(GTest REQUIRED) include_directories(${GTEST_INCLUDE_DIRS}) - -find_library( - hsa-amd-aqlprofile64 - REQUIRED - NAMES hsa-amd-aqlprofile64 hsa-amd-aqlprofile - HINTS /opt/rocm ${CMAKE_BINARY_DIR} ${CMAKE_INSTALL_PREFIX} - PATHS /opt/rocm - PATH_SUFFIXES lib) +if(NOT DEFINED AQLPROFILE_TARGET) + find_library( + AQLPROFILE_TARGET + REQUIRED + NAMES hsa-amd-aqlprofile64 hsa-amd-aqlprofile + HINTS /opt/rocm ${CMAKE_INSTALL_PREFIX} + PATHS /opt/rocm + PATH_SUFFIXES lib) +endif() # Add test for memory manager add_executable(gfx9-memory-manager-test) @@ -105,7 +106,7 @@ target_link_libraries( counters-test PRIVATE hsa-runtime64::hsa-runtime64 - ${hsa-amd-aqlprofile64} + ${AQLPROFILE_TARGET} GTest::gtest GTest::gtest_main GTest::gmock diff --git a/projects/aqlprofile/test/integration/CMakeLists.txt b/projects/aqlprofile/test/integration/CMakeLists.txt index 2df0ecbc93..7c620b8acf 100644 --- a/projects/aqlprofile/test/integration/CMakeLists.txt +++ b/projects/aqlprofile/test/integration/CMakeLists.txt @@ -3,12 +3,15 @@ project(Aqlprofile_v2_tests LANGUAGES C CXX HIP) set(CMAKE_CXX_STANDARD 17) -find_library( - hsa-amd-aqlprofile64 - REQUIRED - NAMES hsa-amd-aqlprofile64 hsa-amd-aqlprofile64 - HINTS /opt/rocm/lib /opt/rocm - PATHS /opt/rocm/lib /opt/rocm) +if(NOT DEFINED AQLPROFILE_TARGET) + find_library( + AQLPROFILE_TARGET + REQUIRED + NAMES hsa-amd-aqlprofile64 hsa-amd-aqlprofile + HINTS /opt/rocm ${CMAKE_INSTALL_PREFIX} + PATHS /opt/rocm + PATH_SUFFIXES lib) +endif() find_package( hsa-runtime64 @@ -51,13 +54,13 @@ set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g -O1" ) add_library(intercept SHARED) target_include_directories(intercept PRIVATE ${HSA_RUNTIME_INC_PATH} ${CMAKE_CURRENT_SOURCE_DIR} ${CMAKE_SOURCE_DIR}/src/core/include/ /opt/rocm/include) target_sources(intercept PRIVATE intercept.cpp) -target_link_libraries(intercept PRIVATE hsa-runtime64::hsa-runtime64 ${hsa-amd-aqlprofile64}) +target_link_libraries(intercept PRIVATE hsa-runtime64::hsa-runtime64 ${AQLPROFILE_TARGET}) target_link_options(intercept PRIVATE -Wl,--version-script=${CMAKE_CURRENT_SOURCE_DIR}/exportmap -Wl,--no-undefined) target_compile_definitions(intercept PUBLIC AMD_INTERNAL_BUILD) add_executable(testv2) target_sources(testv2 PRIVATE main.cpp workload.cpp counter.cpp agent.cpp) target_include_directories(testv2 PRIVATE ${CMAKE_CURRENT_SOURCE_DIR} ${CMAKE_SOURCE_DIR}/src/core/include/ ${HSA_RUNTIME_INC_PATH} /opt/rocm/include) -target_link_libraries(testv2 PRIVATE hsa-runtime64::hsa-runtime64 ${hsa-amd-aqlprofile64}) +target_link_libraries(testv2 PRIVATE hsa-runtime64::hsa-runtime64 ${AQLPROFILE_TARGET}) target_compile_definitions(testv2 PUBLIC AMD_INTERNAL_BUILD) # Add a PRELOAD environment with libintercept