Browse Source

modify xpu infer bug (#4006)

a31413510 6 months ago
parent
commit
1cd127e44c
1 changed files with 2 additions and 0 deletions
  1. 2 0
      paddlex/inference/models/common/static_infer.py

+ 2 - 0
paddlex/inference/models/common/static_infer.py

@@ -423,6 +423,8 @@ class PaddleInfer(StaticInfer):
                     config.enable_new_ir(self._option.enable_new_ir)
                 if hasattr(config, "enable_new_executor"):
                     config.enable_new_executor()
+                config.delete_pass("conv2d_bn_xpu_fuse_pass")
+                config.delete_pass("transfer_layout_pass")
             elif self._option.device_type == "mlu":
                 config.enable_custom_device("mlu", self._option.device_id)
                 if hasattr(config, "enable_new_ir"):