Fixing setting of GPUs to 2 when 1 or less GPUs on system for unit tests (#481)

이 커밋은 다음에 포함됨:
Stanley Tsang 작성하고 GitHub가 2021-12-09 11:04:31 -07:00를 커밋함
1 부모 a94b953bcc
커밋 bbbb35ceec
1 파일이 변경됨
+1 -1
+1 -1
파일 보기
@@ -70,7 +70,7 @@ if(BUILD_TESTS)
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)
if(${gtest_num_gpus} MATCHES "0" OR ${gtest_num_gpus} MATCHES "1")
set(gtest_num_gpus,"2")
set(gtest_num_gpus "2")
endif()
target_compile_options(UnitTests PRIVATE -DGTESTS_NUM_GPUS=${gtest_num_gpus})