Skip to content

Commit

Permalink
small fix: Remove extraneous argument in compile (#2635)
Browse files Browse the repository at this point in the history
  • Loading branch information
gs-olive authored Feb 5, 2024
1 parent a39d254 commit 74683c1
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions py/torch_tensorrt/dynamo/_compiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,13 +81,11 @@ def compile(
engine_capability: EngineCapability = ENGINE_CAPABILITY,
refit: bool = REFIT,
debug: bool = DEBUG,
capability: EngineCapability = EngineCapability.default,
num_avg_timing_iters: int = NUM_AVG_TIMING_ITERS,
workspace_size: int = WORKSPACE_SIZE,
dla_sram_size: int = DLA_SRAM_SIZE,
dla_local_dram_size: int = DLA_LOCAL_DRAM_SIZE,
dla_global_dram_size: int = DLA_GLOBAL_DRAM_SIZE,
calibrator: object = None,
truncate_long_and_double: bool = TRUNCATE_LONG_AND_DOUBLE,
require_full_compilation: bool = REQUIRE_FULL_COMPILATION,
min_block_size: int = MIN_BLOCK_SIZE,
Expand Down Expand Up @@ -168,6 +166,12 @@ def compile(
if debug:
set_log_level(logger.parent, logging.DEBUG)

if torch_executed_modules is not None and torch_executed_modules:
logger.warning(
f"Detected torch_executed_modules was non-empty: {torch_executed_modules}"
"\nThis feature is unimplemented in Torch-TRT Dynamo currently."
)

if not isinstance(inputs, collections.abc.Sequence):
inputs = [inputs]

Expand Down Expand Up @@ -226,6 +230,7 @@ def compile(
"use_python_runtime": use_python_runtime,
"truncate_long_and_double": truncate_long_and_double,
"use_fast_partitioner": use_fast_partitioner,
"num_avg_timing_iters": num_avg_timing_iters,
"enable_experimental_decompositions": enable_experimental_decompositions,
"require_full_compilation": require_full_compilation,
"disable_tf32": disable_tf32,
Expand Down

0 comments on commit 74683c1

Please sign in to comment.