Skip to content

Commit b3ef68e

Browse files
peri044zewenli98
andauthored
fix: l2_limit_for_tiling (#3479) cherry pick to 2.7 (#3493)
Co-authored-by: Zewen (Evan) Li <zewenl@nvidia.com>
1 parent 7680cb1 commit b3ef68e

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -345,9 +345,10 @@ def _populate_trt_builder_config(
345345
self.compilation_settings.tiling_optimization_level
346346
]
347347

348-
builder_config.l2_limit_for_tiling = (
349-
self.compilation_settings.l2_limit_for_tiling
350-
)
348+
if self.compilation_settings.l2_limit_for_tiling != -1:
349+
builder_config.l2_limit_for_tiling = (
350+
self.compilation_settings.l2_limit_for_tiling
351+
)
351352

352353
return builder_config
353354

0 commit comments

Comments
 (0)