diff --git a/deploy/pptracking/python/det_infer.py b/deploy/pptracking/python/det_infer.py index a161962b8ef2d4f2c41bb5acdd02b92850d07df1..c157392336a21ea516280bdd1953c61573952e4b 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 752a07efa48d688fbde2523253a953b28060a879..77f509c2408fccb39eb93c1f9df8105717c9de97 100644 --- a/deploy/python/infer.py +++ b/deploy/python/infer.py @@ -515,6 +515,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 9b606aa44bba02508ad9a3de8e4923cc76f490ca..d3afb138ef089581de630257d10e419578df97b4 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()