Enabling gdrcopy option for gfx950 (#1955)

[ROCm/rccl commit: fedddb452c]
Cette révision appartient à :
gilbertlee-amd a créé et GitHub a révisé 2025-10-15 10:55:25 -06:00
1 Parent 455d516dc4
révision bb85692891
1 fichier modifié
+2 -1
+2 -1
Voir le fichier
@@ -169,7 +169,8 @@ static gdr_t ncclGdrInit() {
return NULL;
}
GcnArchNameFormat(devProp.gcnArchName, gcnArchNameSubstr);
if (IsArchMatch(gcnArchNameSubstr, "gfx942")) {
if (IsArchMatch(gcnArchNameSubstr, "gfx942") ||
IsArchMatch(gcnArchNameSubstr, "gfx950")) {
INFO(NCCL_INIT, "Enabled GDRCopy equivalent memory allocation on %s", gcnArchNameSubstr);
return (gdr_t)0x12345678L;
} else {