Add missing gfx942 support (#927)

このコミットが含まれているのは:
Wenkai Du
2023-10-23 12:04:37 -07:00
committed by GitHub
コミット c4e65fd382
+3
ファイルの表示
@@ -380,7 +380,10 @@ ncclResult_t ncclTopoAddGpu(struct ncclXmlNode* xmlGpu, struct ncclTopoSystem* s
gpu->gpu.gcn = "gfx940";
} else if (strcmp(gpu->gpu.gcn, "941") == 0) {
gpu->gpu.gcn = "gfx941";
} else if (strcmp(gpu->gpu.gcn, "942") == 0) {
gpu->gpu.gcn = "gfx942";
}
rcclHipDeviceArch_t arch;
NCCLCHECK(xmlGetAttrInt(xmlGpu, "arch", &arch.value));
memcpy(&gpu->gpu.arch, &arch.arch, sizeof(hipDeviceArch_t));