Enabling gdrcopy option for gfx950 (#1955)

此提交包含在:
gilbertlee-amd
2025-10-15 10:55:25 -06:00
提交者 GitHub
父節點 c70f5b4621
當前提交 fedddb452c
+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 {