1 父節點
455d516dc4
當前提交
bb85692891
1 file changed
+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 {
|
||||
|
||||
Reference in new issue
封鎖使用者