From d880297f1eee4c8559e5d2d73d8938fba97e478b Mon Sep 17 00:00:00 2001 From: "Zewen (Evan) Li" Date: Sat, 26 Apr 2025 17:12:09 -0700 Subject: [PATCH] fix: l2_limit_for_tiling (#3479) --- py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py b/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py index d886efa1a2..d69536d907 100644 --- a/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py +++ b/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py @@ -345,9 +345,10 @@ def _populate_trt_builder_config( self.compilation_settings.tiling_optimization_level ] - builder_config.l2_limit_for_tiling = ( - self.compilation_settings.l2_limit_for_tiling - ) + if self.compilation_settings.l2_limit_for_tiling != -1: + builder_config.l2_limit_for_tiling = ( + self.compilation_settings.l2_limit_for_tiling + ) return builder_config