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 <Xiaogang.Chen@amd.com>
Change-Id: I800042f93d4e0955040f450e9b5cf09927971323


[ROCm/ROCR-Runtime commit: 665e6f80bf]
This commit is contained in:
Xiaogang Chen
2024-12-13 01:42:06 -06:00
کامیت شده توسط Xiaogang Chen
والد b8279a6de1
کامیت 7ca6b20d3f
@@ -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();