torch.cuda.gds.gds_register_buffer

torch.cuda.gds.gds_register_buffer(s)

將 CUDA 裝置上的 storage 註冊為 cufile 緩衝區。

示例

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

引數

s (Storage) – 要註冊的緩衝區。