Skip to content

Commit 74683c1

Browse files
authored
small fix: Remove extraneous argument in compile (#2635)
1 parent a39d254 commit 74683c1

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

py/torch_tensorrt/dynamo/_compiler.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,13 +81,11 @@ def compile(
8181
engine_capability: EngineCapability = ENGINE_CAPABILITY,
8282
refit: bool = REFIT,
8383
debug: bool = DEBUG,
84-
capability: EngineCapability = EngineCapability.default,
8584
num_avg_timing_iters: int = NUM_AVG_TIMING_ITERS,
8685
workspace_size: int = WORKSPACE_SIZE,
8786
dla_sram_size: int = DLA_SRAM_SIZE,
8887
dla_local_dram_size: int = DLA_LOCAL_DRAM_SIZE,
8988
dla_global_dram_size: int = DLA_GLOBAL_DRAM_SIZE,
90-
calibrator: object = None,
9189
truncate_long_and_double: bool = TRUNCATE_LONG_AND_DOUBLE,
9290
require_full_compilation: bool = REQUIRE_FULL_COMPILATION,
9391
min_block_size: int = MIN_BLOCK_SIZE,
@@ -168,6 +166,12 @@ def compile(
168166
if debug:
169167
set_log_level(logger.parent, logging.DEBUG)
170168

169+
if torch_executed_modules is not None and torch_executed_modules:
170+
logger.warning(
171+
f"Detected torch_executed_modules was non-empty: {torch_executed_modules}"
172+
"\nThis feature is unimplemented in Torch-TRT Dynamo currently."
173+
)
174+
171175
if not isinstance(inputs, collections.abc.Sequence):
172176
inputs = [inputs]
173177

@@ -226,6 +230,7 @@ def compile(
226230
"use_python_runtime": use_python_runtime,
227231
"truncate_long_and_double": truncate_long_and_double,
228232
"use_fast_partitioner": use_fast_partitioner,
233+
"num_avg_timing_iters": num_avg_timing_iters,
229234
"enable_experimental_decompositions": enable_experimental_decompositions,
230235
"require_full_compilation": require_full_compilation,
231236
"disable_tf32": disable_tf32,

0 commit comments

Comments
 (0)