gfx12 Disable ll protocol (#1268)

[ROCm/rccl commit: c246e25f8e]
このコミットが含まれているのは:
akolliasAMD
2024-07-26 08:59:55 -06:00
committed by GitHub
コミット 37c44d531b
5個のファイルの変更、33行の追加、23行の削除
+9 -7
ファイルの表示
@@ -15,7 +15,7 @@ namespace RcclUnitTesting
int const UT_SINGLE_PROCESS = (1<<0);
int const UT_MULTI_PROCESS = (1<<1);
int getArchInfo(bool *isRightArch)
int getArchInfo(bool *isRightArch, const char *gfx)
{
// Prepare parent->child pipe
int pipefd[2];
@@ -25,7 +25,7 @@ namespace RcclUnitTesting
}
pid_t pid = fork();
if (0 == pid) {
bool isGfx94 = false;
bool isGfxTest = false;
int dev;
hipGetDeviceCount(&dev);
for (int deviceId = 0; deviceId < dev; deviceId++) {
@@ -34,14 +34,14 @@ namespace RcclUnitTesting
hipGetDeviceProperties(&devProp, deviceId);
char *gcnArchNameToken = strtok(devProp.gcnArchName, ":");
strcpy(gcn, gcnArchNameToken);
if(std::strncmp("gfx94", gcn, 5) == 0) {
isGfx94 = true;
if(std::strncmp(gfx, gcn, 5) == 0) {
isGfxTest = true;
} else {
isGfx94 = false;
isGfxTest = false;
break;
}
}
if (write(pipefd[1], &isGfx94, sizeof(isGfx94)) != sizeof(isGfx94)) return TEST_FAIL;
if (write(pipefd[1], &isGfxTest, sizeof(isGfxTest)) != sizeof(isGfxTest)) return TEST_FAIL;
close(pipefd[0]);
close(pipefd[1]);
exit(EXIT_SUCCESS);
@@ -95,7 +95,9 @@ namespace RcclUnitTesting
numDetectedGpus = 0;
getDeviceCount(&numDetectedGpus);
isGfx94 = false;
getArchInfo(&isGfx94);
getArchInfo(&isGfx94, "gfx94");
isGfx12 = false;
getArchInfo(&isGfx12, "gfx12");
showNames = GetEnvVar("UT_SHOW_NAMES" , 1);
minGpus = GetEnvVar("UT_MIN_GPUS" , 2);