From f5f778344a4b8d3cc858b191bde506e028f5e7a1 Mon Sep 17 00:00:00 2001 From: Feng Ni Date: Thu, 9 Dec 2021 09:44:33 +0800 Subject: [PATCH] fix xpu deploy infer enable_lite_engine (#4843) --- deploy/pptracking/python/det_infer.py | 1 + deploy/python/infer.py | 1 + static/deploy/python/infer.py | 1 + 3 files changed, 3 insertions(+) diff --git a/deploy/pptracking/python/det_infer.py b/deploy/pptracking/python/det_infer.py index a161962b8..c15739233 100644 --- a/deploy/pptracking/python/det_infer.py +++ b/deploy/pptracking/python/det_infer.py @@ -410,6 +410,7 @@ def load_predictor(model_dir, # optimize graph and fuse op config.switch_ir_optim(True) elif device == 'XPU': + config.enable_lite_engine() config.enable_xpu(10 * 1024 * 1024) else: config.disable_gpu() diff --git a/deploy/python/infer.py b/deploy/python/infer.py index 4145ff6b1..c5423af31 100644 --- a/deploy/python/infer.py +++ b/deploy/python/infer.py @@ -511,6 +511,7 @@ def load_predictor(model_dir, # optimize graph and fuse op config.switch_ir_optim(True) elif device == 'XPU': + config.enable_lite_engine() config.enable_xpu(10 * 1024 * 1024) else: config.disable_gpu() diff --git a/static/deploy/python/infer.py b/static/deploy/python/infer.py index 9b606aa44..d3afb138e 100644 --- a/static/deploy/python/infer.py +++ b/static/deploy/python/infer.py @@ -415,6 +415,7 @@ def load_predictor(model_dir, # optimize graph and fuse op config.switch_ir_optim(True) elif device == 'XPU': + config.enable_lite_engine() config.enable_xpu(10 * 1024 * 1024) else: config.disable_gpu() -- GitLab