[ThF%SSKrSSKrSSKJrJr SSKrSSKJr /SQr\ \ \ S'S\ S\4Sjr \ "\RS 5(Gd<\ "\RS 5(ae\ "\RS 5(ae\ "\RS 5(ae\ "\RS 5(ae\ "\RS5(ae\ "S 5\RRS '\ "S 5\RRS '\ "S 5\RRS '\ "S 5\RRS '\ "S 5\RRS '\ "S5\RRS'S\SS4SjrS\SS4Sjr"SS5rg)N)CallableOptional)Storage)gds_register_buffergds_deregister_bufferGdsFile__all__namereturnc^U4SjnU$)Nc">[STS35e)Nz torch._C.z" is not supported on this platform) RuntimeError)argskwargsr s F/var/www/auris/envauris/lib/python3.13/site-packages/torch/cuda/gds.pyfn_dummy_fn..fnsYtf,NOPP)r rs` r _dummy_fnrsQ Ir_gds_register_buffer_gds_deregister_buffer_gds_register_handle_gds_deregister_handle_gds_load_storage_gds_save_storagescB[RRU5 g)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)torch_Crrs rrr&s HH!!!$rcB[RRU5 g)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)rr rr!s rrr6s HH##A&rc|\rSrSrSrS\S\4SjrSSjrSS jr SS jr SS \ S \SS4S jjr SS \ S \SS4Sjjr Srg)rGa7Wrapper 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 filenameflagsc[RS:Xa [S5eXlX l[ R "X[ R-5UlSUl UR5 g)Nwin32z*GdsFile is not supported on this platform.) sysplatformrr%r&osopenO_DIRECTfdhandleregister_handle)selfr%r&s r__init__GdsFile.__init__gsP <<7 "KL L  ''(BKK$78%)  rr Nc~URbUR5 [R"UR5 g)N)r/deregister_handler+closer.r1s r__del__GdsFile.__del__ps( ;; "  " " $ rcURbS5e[RRUR5Ulg)z`Registers file descriptor to cuFile Driver. This is a wrapper around ``cuFileHandleRegister``. Nz4Cannot register a handle that is already registered.)r/rr rr.r7s rr0GdsFile.register_handleus: KK  B A B hh33DGG< rcURcS5e[RRUR5 SUlg)zfDeregisters file descriptor from cuFile Driver. This is a wrapper around ``cuFileHandleDeregister``. Nz2Cannot deregister a handle that is not registered.)r/rr rr7s rr5GdsFile.deregister_handles= KK # @ ? @ # '' 4 rstorageoffsetcURcS5e[RRURX5 g)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/rr rr1r>r?s r load_storageGdsFile.load_storages8 KK # B A B # ""4;;@rcURcS5e[RRURX5 g)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/rr rrAs r save_storageGdsFile.save_storages8 KK # @ ? @ # ""4;;@r)r.r%r&r/)r N)r)__name__ __module__ __qualname____firstlineno____doc__strintr2r8r0r5rrBrE__static_attributes__rrrrrGsk>S =  AG AS A A AG AS A A Arr)r+r)typingrrr torch.typesrr listrL__annotations__rhasattrr __dict__rrrrrrrUs % cCHuxx/00uxx!9::: :uxx!7888 8uxx!9::: :uxx!4555 5uxx!4555 509:P0QEHH,-2;