diff --git a/superbench/benchmarks/micro_benchmarks/tensorrt_inference_performance.py b/superbench/benchmarks/micro_benchmarks/tensorrt_inference_performance.py index 4d5a5b4b7..7b299e45f 100644 --- a/superbench/benchmarks/micro_benchmarks/tensorrt_inference_performance.py +++ b/superbench/benchmarks/micro_benchmarks/tensorrt_inference_performance.py @@ -104,7 +104,7 @@ def _preprocess(self): # build options '--explicitBatch', f'--optShapes=input:{input_shape}', - '--workspace=8192', + '–-memPoolSize=8G', None if self._args.precision == 'fp32' else f'--{self._args.precision}', # inference options f'--iterations={self._args.iterations}',