Skip to content

Commit 21299ae

Browse files
committed
fix: distingush engines based on compilation settings in addition to graph structure
Signed-off-by: Naren Dasan <[email protected]>
1 parent 972b78e commit 21299ae

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

py/torch_tensorrt/dynamo/_compiler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -513,7 +513,7 @@ def convert_exported_program_to_serialized_trt_engine(
513513
require_full_compilation: bool = _defaults.REQUIRE_FULL_COMPILATION,
514514
disable_tf32: bool = _defaults.DISABLE_TF32,
515515
sparse_weights: bool = _defaults.SPARSE_WEIGHTS,
516-
make_refittable: bool = _defaults.MAKE_refittable,
516+
make_refittable: bool = _defaults.MAKE_REFITTABLE,
517517
engine_capability: EngineCapability = _defaults.ENGINE_CAPABILITY,
518518
num_avg_timing_iters: int = _defaults.NUM_AVG_TIMING_ITERS,
519519
dla_sram_size: int = _defaults.DLA_SRAM_SIZE,

py/torch_tensorrt/dynamo/runtime/_MutableTorchTensorRTModule.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def __init__(
6565
Union[torch.dtype, dtype]
6666
] = _defaults.ENABLED_PRECISIONS,
6767
engine_capability: EngineCapability = _defaults.ENGINE_CAPABILITY,
68-
make_refittable: bool = _defaults.MAKE_refittable,
68+
make_refittable: bool = _defaults.MAKE_REFITTABLE,
6969
debug: bool = _defaults.DEBUG,
7070
num_avg_timing_iters: int = _defaults.NUM_AVG_TIMING_ITERS,
7171
workspace_size: int = _defaults.WORKSPACE_SIZE,

0 commit comments

Comments
 (0)