From c1158a6c76bcd341ce8578035bc925ea4253c340 Mon Sep 17 00:00:00 2001 From: Xiaogang Chen Date: Wed, 20 Nov 2024 01:09:01 -0600 Subject: [PATCH] kfdtest: update run_kfdtest.sh when HSA_TEST_GPUS_NUM is enable Signed-off-by: Xiaogang Chen Change-Id: I68e4f4f8fd7397d3b39b3473638836e4531bef2b [ROCm/ROCR-Runtime commit: ec457752dc40ba835cfd4bfd752069f8e9627114] --- .../tests/kfdtest/scripts/run_kfdtest.sh | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/projects/rocr-runtime/libhsakmt/tests/kfdtest/scripts/run_kfdtest.sh b/projects/rocr-runtime/libhsakmt/tests/kfdtest/scripts/run_kfdtest.sh index 0d8c6fb19f..3d2b627d1e 100755 --- a/projects/rocr-runtime/libhsakmt/tests/kfdtest/scripts/run_kfdtest.sh +++ b/projects/rocr-runtime/libhsakmt/tests/kfdtest/scripts/run_kfdtest.sh @@ -234,10 +234,18 @@ runKfdTest() { fi sudo docker rm kfdtest_docker else - echo "" - echo "++++ Starting testing node $hsaNode ($nodeName) ++++" - $GDB $KFDTEST "--node=$hsaNode" $gtestFilter $GTEST_ARGS - echo "---- Finished testing node $hsaNode ($nodeName) ----" + if [ "$HSA_TEST_GPUS_NUM" != "" ]; then + echo "++++ Starting parallel testing on $HSA_TEST_GPUS_NUM gpu(s) ++++" + $GDB $KFDTEST $gtestFilter $GTEST_ARGS + echo "++++ Finished parallel testing on $HSA_TEST_GPUS_NUM gpu(s) ++++" + exit 0; + else + echo "" + echo "++++ Starting testing node $hsaNode ($nodeName) ++++" + $GDB $KFDTEST "--node=$hsaNode" $gtestFilter $GTEST_ARGS + echo "---- Finished testing node $hsaNode ($nodeName) ----" + fi + fi