torch.cuda.gds.gds_deregister_buffer β€” PyTorch 2.7 documentation (original) (raw)

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.