Skip to content

Commit 02ecf05

Browse files
committed
feat: Disable require_full_compilation in torch compile
1 parent b052376 commit 02ecf05

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

py/torch_tensorrt/dynamo/utils.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -164,6 +164,14 @@ def parse_dynamo_kwargs(kwargs: Any) -> CompilationSettings:
164164
# Parse input runtime specification
165165
settings.use_python_runtime = use_python_runtime_parser(settings.use_python_runtime)
166166

167+
# Ignore and warn about require_full_compilation flag
168+
if settings.require_full_compilation:
169+
logger.warning(
170+
"Detected require_full_compilation=True for a torch.compile run. "
171+
"This option has no effect in torch.compile."
172+
)
173+
settings.require_full_compilation = False
174+
167175
logger.debug(f"Compiling with Settings:\n{settings}")
168176

169177
return settings

0 commit comments

Comments
 (0)