Skip to content

Commit 44dfc31

Browse files
committed
[WIP]
1 parent 9c2a4a9 commit 44dfc31

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

paddlex/inference/models/common/static_infer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -358,7 +358,8 @@ def _create(
358358
logging.debug("`device_id` has been set to None")
359359

360360
if (
361-
self._option.device_type in ("gpu", "dcu", "npu", "mlu", "gcu", "xpu", "iluvatar_gpu")
361+
self._option.device_type
362+
in ("gpu", "dcu", "npu", "mlu", "gcu", "xpu", "iluvatar_gpu")
362363
and self._option.device_id is None
363364
):
364365
self._option.device_id = 0
@@ -399,6 +400,7 @@ def _create(
399400
if hasattr(config, "enable_new_executor"):
400401
config.enable_new_executor()
401402
config.set_optimization_level(3)
403+
config.delete_pass("matmul_add_act_fuse_pass")
402404
elif self._option.device_type == "npu":
403405
config.enable_custom_device("npu", self._option.device_id)
404406
if hasattr(config, "enable_new_ir"):

0 commit comments

Comments
 (0)