torch.cuda.gds.gds_deregister_buffer

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

在 CUDA 设备上注销之前已注册的存储作为 cufile 缓冲区。

示例：

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

参数:

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