Enabling gdrcopy option for gfx950 (#1955)
此提交包含在:
+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 {
|
||||
|
||||
新增問題並參考
封鎖使用者