Update runtime.rst

pytorch · Feb 5, 2025 · 3ee73ad · 3ee73ad
1 parent 04b56e7
commit 3ee73ad
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/docsrc/py_api/runtime.rst b/docsrc/py_api/runtime.rst
@@ -13,6 +13,17 @@ Functions
 
 .. autofunction:: set_multi_device_safe_mode
 
+.. autofunction:: enable_cudagraphs
+
+.. autofunction:: get_cudagraphs_mode
+
+.. autofunction:: get_whole_cudagraphs_mode
+
+.. autofunction:: set_cudagraphs_modue
+
+.. autofunction:: enable_pre_allocated_outputs
+
+.. autofunction:: weight_streaming
 
 Classes
 ---------
@@ -23,4 +34,4 @@ Classes
 
 .. autoclass:: PythonTorchTensorRTModule
    :members:
-   :special-members: __init__
+   :special-members: __init__