torch.cuda.gds.gds_deregister_buffer

torch.cuda.gds.gds_deregister_buffer(s)

注销已注册的 CUDA 设备上的存储，将其作为 cufile 缓冲区。

示例

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

参数

s (Storage) – 要注销的缓冲区。