torch.cuda.gds.gds_deregister_buffer
- torch.cuda.gds.gds_deregister_buffer(s)[source][source]
Deregisters a previously registered storage on a CUDA device as a cufile buffer.
Example:
>>> src = torch.randn(1024, device="cuda") >>> s = src.untyped_storage() >>> gds_register_buffer(s) >>> gds_deregister_buffer(s)
- Parameters
s (Storage) – Buffer to register.