P/jVUddlZddlZddlmZmZmZddlZddlmZgZ ee e d<de defdZ e ejds#e 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 <deddfdZdeddfdZGddZy)N)CallableListOptional)Storage__all__namereturncfd}|S)Nc"tdd)Nz torch._C.z" is not supported on this platform) RuntimeError)argskwargsrs Q/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/cuda/gds.pyfnz_dummy_fn..fn sYtf,NOPP)rrs` r _dummy_fnr sQ Ir_gds_register_buffer_gds_deregister_buffer_gds_register_handle_gds_deregister_handle_gds_load_storage_gds_save_storagescBtjj|yzLRegisters a buffer. Args: s (Storage): Buffer to register. N)torch_Crrs rrr"s  HH!!!$rcBtjj|yr)rrrrs rrr+s  HH##A&rcdeZdZdZdedefdZddZddZdd Z dd e d eddfd Z dd e d eddfd Z y)_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 filenameflagsctjdk(r td||_||_t j ||t jz|_d|_ |jy)Nwin32z*GdsFile is not supported on this platform.) sysplatformr r#r$osopenO_DIRECTfdhandleregister_handle)selfr#r$s r__init__z_GdsFile.__init__BsV <<7 "KL L   ''(EBKK$78%)  rr Ncz|j|jtj|jy)N)r-deregister_handler)closer,r/s r__del__z_GdsFile.__del__Ks( ;; "  " " $ rc|jJdtjj|j|_y)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: KK  B A B hh33DGG< rc|jJdtjj|jd|_y)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= KK # @ ? @ # '' 4 rstorageoffsetc|jJdtjj|j||y)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: KK # B A B # ""4;;@rc|jJdtjj|j||y)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: KK # @ ? @ # ""4;;@r)r N)r) __name__ __module__ __qualname____doc__strintr0r5r.r2rr<r>rrrr"r"4se S =  AG AS A A AG AS A Arr")r)r'typingrrrr torch.typesrrrC__annotations__rhasattrr__dict__rrr"rrrrJs ++ cCHuxx/0uxx!9:: :uxx!788 8uxx!9:: :uxx!455 5uxx!455 509:P0QEHH,-2;