+ jpaRt0t^RIt^RIt^RIHt^RIt^RIHt.ROt]^kRRlt RRlt ] !4'Eg`] !]PR 4'd ] !R 4h] !]PR 4'd ] !R 4h] !]PR 4'd ] !R4h] !]PR4'd ] !R4h] !]PR4'd ] !R4h] !R4]PPR&] !R 4]PPR &] !R 4]PPR &] !R 4]PPR &] !R4]PPR&] !R4]PPR&RRltRRlt!RR4tR#)N)CallableStorageGdsFilec0V^8dQhR\R\/#)namereturn)strr)formats"O/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/cuda/gds.py __annotate__rsCHcaV3RlpV#)c"<\RS R24h)z torch._C.z" is not supported on this platform) RuntimeError)argskwargsr s*,r fn_dummy_fn..fnsYtf,NOPPr)r rsf r _dummy_fnrsQ Irc$V^8dQhR\/#)rr )bool)r s"r rrsdrc6\PP#)zReturn ``True`` if GDS (GPUDirect Storage) support is built in. This requires PyTorch to be built with cuFile (CUDA) or hipFile (ROCm). )torch_C_has_gdsrrr is_availablers 88  r_gds_deregister_bufferz?_gds_deregister_buffer exists but _gds_register_buffer does not_gds_register_handlez=_gds_register_handle exists but _gds_register_buffer does not_gds_deregister_handlez?_gds_deregister_handle exists but _gds_register_buffer does not_gds_load_storagez:_gds_load_storage exists but _gds_register_buffer does not_gds_save_storagez:_gds_save_storage exists but _gds_register_buffer does not_gds_register_bufferc(V^8dQhR\RR/#rsr Nr)r s"r rr?s % %7 %t %rcD\PPV4R#)aRegisters a storage on a CUDA device as a cufile buffer. Example:: >>> # xdoctest: +SKIP("gds filesystem requirements") >>> src = torch.randn(1024, device="cuda") >>> s = src.untyped_storage() >>> gds_register_buffer(s) Args: s (Storage): Buffer to register. N)rrr%r(s&r gds_register_bufferr+?s HH!!!$rc(V^8dQhR\RR/#r'r)r s"r rrOs''W''rcD\PPV4R#)aSDeregisters a previously registered storage on a CUDA device as a cufile buffer. Example:: >>> # xdoctest: +SKIP("gds filesystem requirements") >>> src = torch.randn(1024, device="cuda") >>> s = src.untyped_storage() >>> gds_register_buffer(s) >>> gds_deregister_buffer(s) Args: s (Storage): Buffer to register. N)rrr r*s&r gds_deregister_bufferr.Os HH##A&rca]tRt^`toRtV3RlRltV3RlRltV3RlRltV3RlR ltRV3R lR llt RV3R lR llt Rt Vt R#)ra7Wrapper around cuFile. cuFile is a file-like interface to the GPUDirect Storage (GDS) API. See the `cufile docs `_ for more details. Args: filename (str): Name of the file to open. flags (int): Flags to pass to ``os.open`` when opening the file. ``os.O_DIRECT`` will be added automatically. Example:: >>> # xdoctest: +SKIP("gds filesystem requirements") >>> src1 = torch.randn(1024, device="cuda") >>> src2 = torch.randn(2, 1024, device="cuda") >>> file = torch.cuda.gds.GdsFile(f, os.O_CREAT | os.O_RDWR) >>> file.save_storage(src1.untyped_storage(), offset=0) >>> file.save_storage(src2.untyped_storage(), offset=src1.nbytes) >>> dest1 = torch.empty(1024, device="cuda") >>> dest2 = torch.empty(2, 1024, device="cuda") >>> file.load_storage(dest1.untyped_storage(), offset=0) >>> file.load_storage(dest2.untyped_storage(), offset=src1.nbytes) >>> torch.equal(src1, dest1) True >>> torch.equal(src2, dest2) True c&<V^8dQhRS[RS[/#)rfilenameflags)r int)r __classdict__s"r rGdsFile.__annotate__sSrc\PR8Xd \R4hWnW n\ P !W\ P,4VnRVn VP4R#)win32z*GdsFile is not supported on this platform.N) sysplatformrr1r2osopenO_DIRECTfdhandleregister_handle)selfr1r2s&&&r __init__GdsFile.__init__sP <<7 "KL L  ''(BKK$78"&  rc<V^8dQhRR/#rr Nr)r r4s"r rr5srcVPeVP4\P!VP4R#)N)r>deregister_handler:closer=r@s&r __del__GdsFile.__del__s( ;; "  " " $ rc<V^8dQhRR/#rDr)r r4s"r rr5s===rc VPe \R4h\PP VP 4VnR#)z`Registers file descriptor to cuFile Driver. This is a wrapper around ``cuFileHandleRegister``. Nz4Cannot register a handle that is already registered.)r>AssertionErrorrrr!r=rHs&r r?GdsFile.register_handles4 ;; " !WX Xhh33DGG< rc<V^8dQhRR/#rDr)r r4s"r rr5s4rc VPf \R4h\PP VP4RVnR#)zfDeregisters file descriptor from cuFile Driver. This is a wrapper around ``cuFileHandleDeregister``. Nz2Cannot deregister a handle that is not registered.)r>rMrrr"rHs&r rFGdsFile.deregister_handles7 ;;  !UV V '' 4 rc*<V^8dQhRS[RS[RR/#rstorageoffsetr Nrr3)r r4s"r rr5' A AG AS A Arc VPf \R4h\PP VPW4R#)a=Loads data from the file into the storage. This is a wrapper around ``cuFileRead``. ``storage.nbytes()`` of data will be loaded from the file at ``offset`` into the storage. Args: storage (Storage): Storage to load data into. offset (int, optional): Offset into the file to start loading from. (Default: 0) Nz4Cannot load data from a file that is not registered.)r>rMrrr#r@rTrUs&&&r load_storageGdsFile.load_storages2 ;;  !WX X ""4;;@rc*<V^8dQhRS[RS[RR/#rSrV)r r4s"r rr5rWrc VPf \R4h\PP VPW4R#)a%Saves data from the storage into the file. This is a wrapper around ``cuFileWrite``. All bytes of the storage will be written to the file at ``offset``. Args: storage (Storage): Storage to save data from. offset (int, optional): Offset into the file to start saving to. (Default: 0) Nz2Cannot save data to a file that is not registered.)r>rMrrr$rYs&&&r save_storageGdsFile.save_storages2 ;;  !UV V ""4;;@r)r=r1r2r>N)r) __name__ __module__ __qualname____firstlineno____doc__rArIr?rFrZr^__static_attributes____classdictcell__)r4s@r rr`sP> == A A A A ArcVV^8dQh/^\9d\\,;R&#)r__all__)__conditional_annotations__listr )r s"r rrs c r)rr+r.r)rir:r8collections.abcrr torch.typesrrhrrhasattrrrM__dict__r+r.rr)ris@r ros $ ~~uxx122 M  uxx/00 K  uxx122 M  uxx,-- H  uxx,-- H  1::P0QEHH,-2;