Update prototype_source/gpu_direct_storage.py

svekars · mikaylagawarecki · commit 30082cf13ac8 · 2025-05-01T11:20:33.000-07:00
diff --git a/prototype_source/gpu_direct_storage.py b/prototype_source/gpu_direct_storage.py
@@ -31,7 +31,7 @@
 
 ################################################################################
 # Using GPUDirect Storage with ``torch.save`` and ``torch.load``
-# =============================================================
+# ------------------------------------------------------------------------------------
 # GPUDirect Storage requires a storage alignment of 4KB. You can toggle this by using
 # ``torch.utils.serialization.config.save.storage_alignment``:
 
