[inference][ut]update trt workspace size for inference predictor ut (#50439)

* update trt workspace size for inference predictor ut

[inference][ut]update trt workspace size for inference predictor ut (#50439)
* update trt workspace size for inference predictor ut
cf48d20f · Zhang Jun · GitHub · 340931fb · cf48d20f
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/tests/unittests/ir/inference/test_trt_inference_predictor.py ...ts/unittests/ir/inference/test_trt_inference_predictor.py +1 -1

未找到文件。
--- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_inference_predictor.py
+++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_inference_predictor.py
@@ -127,7 +127,7 @@ class BackendPaddle:
                        "shape"
                    ][self.args.test_num][0]
                config.enable_tensorrt_engine(
-                    workspace_size=1 << 33,
+                    workspace_size=1 << 25,
                    precision_mode=precision_mode,
                    max_batch_size=max_batch_size,
                    min_subgraph_size=self.args.subgraph_size,