Skip to content

Commit fefa85a

Browse files
committed
Improving the option logging
1 parent 464b4eb commit fefa85a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

py/torch_tensorrt/dynamo/_settings.py

Lines changed: 1 addition & 1 deletion
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): Using aot_autograd to trace the graph. This is enabled when DTensors or distributed tensors are present in distributed model
100100
"""
101101

102102
enabled_precisions: Set[dtype] = field(default_factory=lambda: ENABLED_PRECISIONS)

0 commit comments

Comments
 (0)