Enabling gdrcopy option for gfx950 (#1955)

[ROCm/rccl commit: fedddb452c]
Этот коммит содержится в:
gilbertlee-amd
2025-10-15 10:55:25 -06:00
коммит произвёл GitHub
родитель 455d516dc4
Коммит bb85692891
+2 -1
Просмотреть файл
@@ -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 {