Skip to content

Commit 2bb0894

Browse files
committed
Improving the option logging
1 parent b44e4c4 commit 2bb0894

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
@@ -92,7 +92,7 @@ class CompilationSettings:
9292
enable_weight_streaming (bool): Enable weight streaming.
9393
enable_cross_compile_for_windows (bool): By default this is False means TensorRT engines can only be executed on the same platform where they were built.
9494
True will enable cross-platform compatibility which allows the engine to be built on Linux and run on Windows
95-
use_distributed_mode_trace (bool): Use aot_export_joint_simple for models, else wrap backend with AOT_autograd if this is True as required for distributed tensors
95+
use_distributed_mode_trace (bool): Using aot_autograd to trace the graph. Enable this only if the model includes distributed operations
9696
"""
9797

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

0 commit comments

Comments
 (0)