torch.cuda.gds.gds_register_buffer

torch.cuda.gds.gds_register_buffer(s)

Registers a storage on a CUDA device as a GDS buffer.

This is a wrapper around cuFileBufRegister (CUDA) / hipFileBufRegister (ROCm).

Example:

>>> src = torch.randn(1024, device="cuda")
>>> s = src.untyped_storage()
>>> gds_register_buffer(s)

Parameters:

s (Storage) – Buffer to register.