[doc] Update options documentation for torch.compile (#2834)

pytorch · May 24, 2024 · 36ed711 · 36ed711
1 parent f22fa4c
commit 36ed711
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/docsrc/dynamo/torch_compile.rst b/docsrc/dynamo/torch_compile.rst
@@ -32,13 +32,14 @@ Customizeable Settings
 
 Custom Setting Usage
 ^^^^^^^^^^^^^^^^^
+
 .. code-block:: python
 
     import torch_tensorrt
     ...
     optimized_model = torch.compile(model, backend="torch_tensorrt", dynamic=False,
                                     options={"truncate_long_and_double": True,
-                                             "precision": torch.half,
+                                             "enabled_precisions": {torch.float, torch.half},
                                              "debug": True,
                                              "min_block_size": 2,
                                              "torch_executed_ops": {"torch.ops.aten.sub.Tensor"},