Enabling gdrcopy option for gfx950 (#1955)

[ROCm/rccl commit: fedddb452c]
Este commit está contenido en:
gilbertlee-amd
2025-10-15 10:55:25 -06:00
cometido por GitHub
padre 455d516dc4
commit bb85692891
+2 -1
Ver fichero
@@ -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 {