Skip to content

Commit 3ee73ad

Browse files
authored
Update runtime.rst
1 parent 04b56e7 commit 3ee73ad

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

docsrc/py_api/runtime.rst

+12-1
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,17 @@ Functions
1313

1414
.. autofunction:: set_multi_device_safe_mode
1515

16+
.. autofunction:: enable_cudagraphs
17+
18+
.. autofunction:: get_cudagraphs_mode
19+
20+
.. autofunction:: get_whole_cudagraphs_mode
21+
22+
.. autofunction:: set_cudagraphs_modue
23+
24+
.. autofunction:: enable_pre_allocated_outputs
25+
26+
.. autofunction:: weight_streaming
1627

1728
Classes
1829
---------
@@ -23,4 +34,4 @@ Classes
2334

2435
.. autoclass:: PythonTorchTensorRTModule
2536
:members:
26-
:special-members: __init__
37+
:special-members: __init__

0 commit comments

Comments
 (0)