AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Unit tests: fix number of GPU detection (#484)

Browse Source 
[ROCm/rccl commit: 6268b87c16]
This commit is contained in:
Wenkai Du
2022-01-05 15:06:12 -08:00
committed by GitHub
parent e5ff692673
commit b5bfb7366a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/test/CMakeLists.txt
+1 -1
View File
@@ -69,7 +69,7 @@ if(BUILD_TESTS)
find_program( rocminfo_executable rocminfo ) find_program( rocminfo_executable rocminfo )
execute_process(COMMAND bash "-c" "${rocminfo_executable} | grep 'Device Type' | grep GPU | wc -l | tr -d '\n'" OUTPUT_VARIABLE gtest_num_gpus) execute_process(COMMAND bash "-c" "${rocminfo_executable} | grep 'Device Type' | grep GPU | wc -l | tr -d '\n'" OUTPUT_VARIABLE gtest_num_gpus)
if(${gtest_num_gpus} MATCHES "0" OR ${gtest_num_gpus} MATCHES "1") if(${gtest_num_gpus} EQUAL "0" OR ${gtest_num_gpus} EQUAL "1")
set(gtest_num_gpus "2") set(gtest_num_gpus "2")
endif() endif()
target_compile_options(UnitTests PRIVATE -DGTESTS_NUM_GPUS=${gtest_num_gpus}) target_compile_options(UnitTests PRIVATE -DGTESTS_NUM_GPUS=${gtest_num_gpus})