Skip to content

Commit 464b4eb

Browse files
committed
Correcting the option in _settings.py and improving the error logging
1 parent 6410087 commit 464b4eb

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

py/torch_tensorrt/dynamo/_settings.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ class CompilationSettings:
9696
True will enable cross-platform compatibility which allows the engine to be built on Linux and run on Windows
9797
tiling_optimization_level (str): The optimization level of tiling strategies. A higher level allows TensorRT to spend more time searching for better tiling strategy. We currently support ["none", "fast", "moderate", "full"].
9898
l2_limit_for_tiling (int): The target L2 cache usage limit (in bytes) for tiling optimization (default is -1 which means no limit).
99-
USE_DISTRIBUTED_MODE_TRACE (bool): Use aot_export_joint_simple, else wrap backend with AOT_autograd, required for distributed tensors
99+
use_distributed_mode_trace (bool): Use aot_export_joint_simple, else wrap backend with AOT_autograd, required for distributed tensors
100100
"""
101101

102102
enabled_precisions: Set[dtype] = field(default_factory=lambda: ENABLED_PRECISIONS)
@@ -137,13 +137,9 @@ class CompilationSettings:
137137
immutable_weights: bool = IMMUTABLE_WEIGHTS
138138
enable_weight_streaming: bool = ENABLE_WEIGHT_STREAMING
139139
enable_cross_compile_for_windows: bool = ENABLE_CROSS_COMPILE_FOR_WINDOWS
140-
<<<<<<< HEAD
141-
use_aot_joint_export: bool = USE_AOT_JOINT_EXPORT
142140
tiling_optimization_level: str = TILING_OPTIMIZATION_LEVEL
143141
l2_limit_for_tiling: int = L2_LIMIT_FOR_TILING
144-
=======
145142
use_distributed_mode_trace: bool = USE_DISTRIBUTED_MODE_TRACE
146-
>>>>>>> 3adafafc4 (restructuring the code to include option use_distributed_mode_trace)
147143

148144

149145
_SETTINGS_TO_BE_ENGINE_INVARIANT = (

0 commit comments

Comments
 (0)