update npu inference api (#9765)

ce657457 · duanyanhui · GitHub · d496c8e8 · ce657457
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

tools/infer/utility.py tools/infer/utility.py +1 -1

未找到文件。
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -246,7 +246,7 @@ def create_predictor(args, mode, logger):
                    logger.info("Please keep your paddlepaddle-gpu >= 2.3.0!")

        elif args.use_npu:
-            config.enable_npu()
+            config.enable_custom_device("npu")
        elif args.use_xpu:
            config.enable_xpu(10 * 1024 * 1024)
        else: