Add missing gfx942 support (#927)

[ROCm/rccl commit: c4e65fd382]
Bu işleme şunda yer alıyor:
Wenkai Du authored and GitHub committed 2023-10-23 12:04:37 -07:00
1 ebeveyn bc7df769a2
işleme cc4de02a86
1 file changed
+3
+3
Dosyayı Görüntüle
@@ -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));