Skip to content

Commit dfae195

Browse files
committed
Remove calibrator and add num_avg_timing_iters
1 parent 43c850a commit dfae195

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

py/torch_tensorrt/dynamo/_compiler.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,6 @@ def compile(
8686
dla_sram_size: int = DLA_SRAM_SIZE,
8787
dla_local_dram_size: int = DLA_LOCAL_DRAM_SIZE,
8888
dla_global_dram_size: int = DLA_GLOBAL_DRAM_SIZE,
89-
calibrator: object = None,
9089
truncate_long_and_double: bool = TRUNCATE_LONG_AND_DOUBLE,
9190
require_full_compilation: bool = REQUIRE_FULL_COMPILATION,
9291
min_block_size: int = MIN_BLOCK_SIZE,
@@ -225,6 +224,7 @@ def compile(
225224
"use_python_runtime": use_python_runtime,
226225
"truncate_long_and_double": truncate_long_and_double,
227226
"use_fast_partitioner": use_fast_partitioner,
227+
"num_avg_timing_iters": num_avg_timing_iters,
228228
"enable_experimental_decompositions": enable_experimental_decompositions,
229229
"require_full_compilation": require_full_compilation,
230230
"disable_tf32": disable_tf32,

0 commit comments

Comments
 (0)