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
والد
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();
|
||||
|
||||
|
||||
مرجع در شماره جدید
Block a user