Skip to content

Commit 42b49f1

Browse files
[doc] Update options documentation for torch.compile (#2834)
1 parent 644a961 commit 42b49f1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

docsrc/dynamo/torch_compile.rst

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,13 +32,14 @@ Customizeable Settings
3232

3333
Custom Setting Usage
3434
^^^^^^^^^^^^^^^^^
35+
3536
.. code-block:: python
3637
3738
import torch_tensorrt
3839
...
3940
optimized_model = torch.compile(model, backend="torch_tensorrt", dynamic=False,
4041
options={"truncate_long_and_double": True,
41-
"precision": torch.half,
42+
"enabled_precisions": {torch.float, torch.half},
4243
"debug": True,
4344
"min_block_size": 2,
4445
"torch_executed_ops": {"torch.ops.aten.sub.Tensor"},

0 commit comments

Comments
 (0)