torch.cuda.gds.gds_register_buffer

torch.cuda.gds.gds_register_buffer(s)[source][source]

在 CUDA 设备上注册存储为 cufile 缓冲区。

示例：

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

参数:

s（存储）- 缓冲区注册。