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

这个提交包含在:
Wenkai Du 编写, GitHub 于 2022-01-05 15:06:12 -08:00 提交
1 父节点 4234a638b5
当前提交 6268b87c16
1 个文件已更改
+1 -1
+1 -1
查看文件
@@ -69,7 +69,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")
if(${gtest_num_gpus} EQUAL "0" OR ${gtest_num_gpus} EQUAL "1")
set(gtest_num_gpus "2")
endif()
target_compile_options(UnitTests PRIVATE -DGTESTS_NUM_GPUS=${gtest_num_gpus})