File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -513,7 +513,7 @@ def convert_exported_program_to_serialized_trt_engine(
513
513
require_full_compilation : bool = _defaults .REQUIRE_FULL_COMPILATION ,
514
514
disable_tf32 : bool = _defaults .DISABLE_TF32 ,
515
515
sparse_weights : bool = _defaults .SPARSE_WEIGHTS ,
516
- make_refittable : bool = _defaults .MAKE_refittable ,
516
+ make_refittable : bool = _defaults .MAKE_REFITTABLE ,
517
517
engine_capability : EngineCapability = _defaults .ENGINE_CAPABILITY ,
518
518
num_avg_timing_iters : int = _defaults .NUM_AVG_TIMING_ITERS ,
519
519
dla_sram_size : int = _defaults .DLA_SRAM_SIZE ,
Original file line number Diff line number Diff line change @@ -65,7 +65,7 @@ def __init__(
65
65
Union [torch .dtype , dtype ]
66
66
] = _defaults .ENABLED_PRECISIONS ,
67
67
engine_capability : EngineCapability = _defaults .ENGINE_CAPABILITY ,
68
- make_refittable : bool = _defaults .MAKE_refittable ,
68
+ make_refittable : bool = _defaults .MAKE_REFITTABLE ,
69
69
debug : bool = _defaults .DEBUG ,
70
70
num_avg_timing_iters : int = _defaults .NUM_AVG_TIMING_ITERS ,
71
71
workspace_size : int = _defaults .WORKSPACE_SIZE ,
You can’t perform that action at this time.
0 commit comments