jUddlZddlZddlmZmZmZddlZddlmZgZ ee e d<de defdZ e ejdse ejd rJe ejd rJe ejd rJe ejd rJe ejd rJe dejjd<e d ejjd <e d ejjd <e d ejjd <e d ejjd <e d ejjd <deddfdZdeddfdZGddZdS)N)CallableListOptional)Storage__all__namereturncfd}|S)Nc*tdd)Nz torch._C.z" is not supported on this platform) RuntimeError)argskwargsrs ?C:\Projects\Historical\venv\Lib\site-packages\torch/cuda/gds.pyfnz_dummy_fn..fn sOtOOOPPP)rrs` r _dummy_fnr s(QQQQQ Ir_gds_register_buffer_gds_deregister_buffer_gds_register_handle_gds_deregister_handle_gds_load_storage_gds_save_storagescDtj|dSzLRegisters a buffer. Args: s (Storage): Buffer to register. N)torch_Crrs rrr"s  H!!!$$$$$rcDtj|dSr)rrrrs rrr+s  H##A&&&&&rcfeZdZdZdedefdZddZddZdd Z dd e d eddfd Z dd e d eddfdZ dS)_GdsFileaWrapper around cuFile. cuFile is a file-like interface to the GPUDirect Storage (GDS) API. 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. .. _CUDA GPUDirect Storage Documentation: https://docs.nvidia.com/gpudirect-storage/api-reference-guide/index.html#cufile-io-api filenameflagsctjdkrtd||_||_t j||t jz|_d|_ | dS)Nwin32z*GdsFile is not supported on this platform.) sysplatformr r#r$osopenO_DIRECTfdhandleregister_handle)selfr#r$s r__init__z_GdsFile.__init__Bse <7 " "KLL L   '(EBK$788%)  rr Ncn|j|tj|jdS)N)r-deregister_handler)closer,r/s r__del__z_GdsFile.__del__Ks4 ; "  " " $ $ $ rcz|j Jdtj|j|_dS)zpRegisters file descriptor to cuFile Driver. This is a wrapper around ``cuFileHandleRegister``. Nz4Cannot register a handle that is already registered.)r-rrrr,r4s rr.z_GdsFile.register_handlePs: K   A  h33DG<< rc~|j Jdtj|jd|_dS)zvDeregisters file descriptor from cuFile Driver. This is a wrapper around ``cuFileHandleDeregister``. Nz2Cannot deregister a handle that is not registered.)r-rrrr4s rr2z_GdsFile.deregister_handleZs@ K # # ? $ # # '' 444 rrstorageoffsetct|j Jdtj|j||dS)amLoads 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-rrrr/r8r9s r load_storagez_GdsFile.load_storagees? K # # A $ # # ""4;@@@@@rct|j Jdtj|j||dS)aUSaves 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-rrrr;s r save_storagez_GdsFile.save_storagets? K # # ? $ # # ""4;@@@@@r)r N)r) __name__ __module__ __qualname____doc__strintr0r5r.r2rr<r>rrrr"r"4s  S ====     A AG AS A A A A A A AG AS A A A A A A Arr")r)r'typingrrrr torch.typesrrrC__annotations__rhasattrr__dict__rrr"rrrrJsW ++++++++++ cCHwux/00 Lwux!9:::: :wux!78888 8wux!9:::: :wux!45555 5wux!45555 509 :P0Q0QEH,-2;)