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":