未验证 提交 d36ba700 编写于 作者: F Feng Ni 提交者: GitHub

[cherry-pick] Fix xpu lite engine (#4848)

* fix plot_tracking_dict, test=document_fix

* fix xpu deploy infer enable_lite_engine, test=document_fix
上级 948ddaa8
...@@ -410,6 +410,7 @@ def load_predictor(model_dir, ...@@ -410,6 +410,7 @@ def load_predictor(model_dir,
# optimize graph and fuse op # optimize graph and fuse op
config.switch_ir_optim(True) config.switch_ir_optim(True)
elif device == 'XPU': elif device == 'XPU':
config.enable_lite_engine()
config.enable_xpu(10 * 1024 * 1024) config.enable_xpu(10 * 1024 * 1024)
else: else:
config.disable_gpu() config.disable_gpu()
......
...@@ -515,6 +515,7 @@ def load_predictor(model_dir, ...@@ -515,6 +515,7 @@ def load_predictor(model_dir,
# optimize graph and fuse op # optimize graph and fuse op
config.switch_ir_optim(True) config.switch_ir_optim(True)
elif device == 'XPU': elif device == 'XPU':
config.enable_lite_engine()
config.enable_xpu(10 * 1024 * 1024) config.enable_xpu(10 * 1024 * 1024)
else: else:
config.disable_gpu() config.disable_gpu()
......
...@@ -415,6 +415,7 @@ def load_predictor(model_dir, ...@@ -415,6 +415,7 @@ def load_predictor(model_dir,
# optimize graph and fuse op # optimize graph and fuse op
config.switch_ir_optim(True) config.switch_ir_optim(True)
elif device == 'XPU': elif device == 'XPU':
config.enable_lite_engine()
config.enable_xpu(10 * 1024 * 1024) config.enable_xpu(10 * 1024 * 1024)
else: else:
config.disable_gpu() config.disable_gpu()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册