We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 04b56e7 commit 3ee73adCopy full SHA for 3ee73ad
docsrc/py_api/runtime.rst
@@ -13,6 +13,17 @@ Functions
13
14
.. autofunction:: set_multi_device_safe_mode
15
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
27
28
Classes
29
---------
@@ -23,4 +34,4 @@ Classes
34
35
.. autoclass:: PythonTorchTensorRTModule
36
:members:
- :special-members: __init__
37
+ :special-members: __init__
0 commit comments