File tree Expand file tree Collapse file tree 2 files changed +14
-1
lines changed Expand file tree Collapse file tree 2 files changed +14
-1
lines changed Original file line number Diff line number Diff line change @@ -270,7 +270,12 @@ def partition(
270
270
271
271
# Construct
272
272
supported_ops = OpSupportTester (torch_executed_ops = torch_executed_ops )
273
- partitioner = TRTPartitioner (gm , supported_ops , min_block_size = min_block_size )
273
+ partitioner = TRTPartitioner (
274
+ gm ,
275
+ supported_ops ,
276
+ min_block_size = min_block_size ,
277
+ require_full_compilation = require_full_compilation ,
278
+ )
274
279
275
280
partitioned_graph = partitioner .partition_graph ()
276
281
Original file line number Diff line number Diff line change @@ -184,6 +184,14 @@ def parse_dynamo_kwargs(kwargs: Any) -> CompilationSettings:
184
184
# Parse input runtime specification
185
185
settings .use_python_runtime = use_python_runtime_parser (settings .use_python_runtime )
186
186
187
+ # Ignore and warn about require_full_compilation flag
188
+ if settings .require_full_compilation :
189
+ logger .warning (
190
+ "Detected require_full_compilation=True for a torch.compile run. "
191
+ "This option has no effect in torch.compile."
192
+ )
193
+ settings .require_full_compilation = False
194
+
187
195
logger .info ("Compilation Settings: %s\n " , settings )
188
196
189
197
return settings
You can’t perform that action at this time.
0 commit comments