wangna11BD 11 місяців тому
батько
коміт
9f445f04a6
1 змінених файлів з 1 додано та 0 видалено
  1. 1 0
      paddlex/utils/device.py

+ 1 - 0
paddlex/utils/device.py

@@ -96,6 +96,7 @@ def set_env_for_device(device):
                 "BKCL_FORCE_SYNC": "1",
                 "BKCL_TIMEOUT": "1800",
                 "FLAGS_use_stride_kernel": "0",
+                "XPU_BLACK_LIST": "pad3d",
             }
             _set(envs)
         if device_type.lower() == "mlu":