Update on "Make torch.cuda.gds APIs public"

mikaylagawarecki · mikaylagawarecki · commit 5768d7fae8c9 · 2025-02-13T20:58:34.000-08:00
Follow up to #145748 that turned USE_CUFILE on for CUDA 12.6 and 12.8 binaries [ghstack-poisoned]
diff --git a/torch/cuda/gds.py b/torch/cuda/gds.py
@@ -38,7 +38,7 @@ def fn(*args, **kwargs):  # type: ignore[no-untyped-def]
 def gds_register_buffer(s: Storage) -> None:
     """Registers a storage on a CUDA device as a cufile buffer.
 
-    .. Example::
+    Example::
 
         >>> # xdoctest: +SKIP("gds filesystem requirements")
         >>> src = torch.randn(1024, device="cuda")
@@ -54,7 +54,7 @@ def gds_register_buffer(s: Storage) -> None:
 def gds_deregister_buffer(s: Storage) -> None:
     """Deregisters a previously registered storage on a CUDA device as a cufile buffer.
 
-    .. Example::
+    Example::
 
         >>> # xdoctest: +SKIP("gds filesystem requirements")
         >>> src = torch.randn(1024, device="cuda")
@@ -81,7 +81,7 @@ class GdsFile:
         flags (int): Flags to pass to ``os.open`` when opening the file. ``os.O_DIRECT`` will
             be added automatically.
 
-    .. Example::
+    Example::
 
         >>> # xdoctest: +SKIP("gds filesystem requirements")
         >>> src1 = torch.randn(1024, device="cuda")
