Browse Source

fix xpu bug for pad3d

wangna11BD 11 tháng trước cách đây
mục cha
commit
9f445f04a6
1 tập tin đã thay đổi với 1 bổ sung0 xóa
  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":