Enabling gdrcopy option for gfx950 (#1955)

[ROCm/rccl commit: fedddb452c]
此提交包含在:
gilbertlee-amd authored and GitHub committed 2025-10-15 10:55:25 -06:00
1 父節點 455d516dc4
當前提交 bb85692891
1 file changed
+2 -1
+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 {