From 7ca6b20d3ffc495349cf0e2f98eebaa3e7022885 Mon Sep 17 00:00:00 2001 From: Xiaogang Chen Date: Fri, 13 Dec 2024 01:42:06 -0600 Subject: [PATCH] kfdtest: Add an env variable to control kfdtest on multi gpu Add HSA_TEST_GPUS_NUM env variable that controls gpu number kfdtest uses. This env variable will be used by following patches. Signed-off-by: Xiaogang Chen Change-Id: I800042f93d4e0955040f450e9b5cf09927971323 [ROCm/ROCR-Runtime commit: 665e6f80bf325db884509a6158bf52b4c177c58f] --- .../libhsakmt/tests/kfdtest/src/KFDTestMain.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/projects/rocr-runtime/libhsakmt/tests/kfdtest/src/KFDTestMain.cpp b/projects/rocr-runtime/libhsakmt/tests/kfdtest/src/KFDTestMain.cpp index 1a806b08f4..4827d58c3b 100644 --- a/projects/rocr-runtime/libhsakmt/tests/kfdtest/src/KFDTestMain.cpp +++ b/projects/rocr-runtime/libhsakmt/tests/kfdtest/src/KFDTestMain.cpp @@ -48,6 +48,7 @@ std::ostream& operator << (std::ostream& out, TESTPROFILE profile) { return out; } +unsigned int g_TestGPUsNum ; unsigned int g_TestRunProfile; unsigned int g_TestENVCaps; unsigned int g_TestTimeOut; @@ -106,6 +107,14 @@ GTEST_API_ int main(int argc, char **argv) { LOG() << "Sleep time in seconds as specified by user: " << std::dec << g_SleepTime << std::endl; } + char *testGPUsNum = NULL; + /* if HSA_TEST_GPUS_NUM is defined use it, otherwise test on 1 gpu */ + testGPUsNum = getenv("HSA_TEST_GPUS_NUM"); + if (testGPUsNum) + g_TestGPUsNum = std::max(1, atoi(testGPUsNum)); + else + g_TestGPUsNum = 1; + /* init LLVM one time*/ Init_LLVM();