Skip to content

Commit a5f98f1

Browse files
one more fix
1 parent ba7f4e8 commit a5f98f1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

prototype_source/gpu_direct_storage.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@
6767
#
6868
# If you are continuously saving the same state dictionary during training, you
6969
# would only need to obtain the offsets once and the same offsets can be re-used. Similarly if tensor is going to
70-
# be loaded to repeatedly one can use the ``torch.cuda.gds.gds_register_buffer`` which wraps
70+
# be saved or loaded to repeatedly one can use the ``torch.cuda.gds.gds_register_buffer`` which wraps
7171
# ``cuFileBufRegister`` to register the storages as gds buffers.
7272

7373

0 commit comments

Comments
 (0)