#hiUddlZddlZddlmZmZmZddlmZddlm Z m Z m Z m ZmZmZmZddlZddlZddlmZddlZddlZddlZ ddlZdZn #e$rdZYnwxYwejZiZe eej fe!d<ed d Z"Gd d Z#dZ$Gddej%j&e#Z'dZ(ee#_)ee#_*ee#_+eddZ,eddZ-dZ.dZ/da0dZ1dZ2d(dZ3dZ4de5fdZ6Gdd Z7ej8e7j)_8ej8e7j*_8ej8e7j+_8Gd!d"e)Z9Gd#d$e7e9%Z:d&e5fd'Z;dS))N)_type_cuda_hpu)Storage)castAnyDictOptionalTypeVarTypeUnion) lru_cacheT_share_memory_mapTz!Union[_StorageBase, TypedStorage])boundcjeZdZUeed<dZeed<dZeed<ej ed<dZ de fdZ d Z d ZdXd ed eedefdZdefdZde fdZde fdZdYdeed edefdZdYdefdZdYdefdZde fdZde fdZde fdZdZdZedeede defdZ edeede defdZ!edeedefdZ"ed d d deedefd!Z#edeedefd"Z$edeedefd#Z%defd$Z&d%Z'de fd&Z(defd'Z)d(Z*d)Z+d*Z,defd+Z-edeedefd,Z.d-Z/ed.Z0e1d/Z2e1d0Z3edefd1Z4edefd2Z5d3Z6deefd4Z7d5Z8d6Z9d7Z:d8Z;d9Zd<Z?d=Z@d>ZAd?ZBd@ZCdAZDdBZEdCZFdDZGdEZ dFZHdGZIdHZJdIZdJZKdKZLdLZMdMZNdNZOdZdePeej ffdPZQdZdePeej ffdQZRdRZSedSdTdUZTdVZUdWZVxZWS)[ _StorageBase_cdataF is_sparse is_sparse_csrdevicecdSNselfargskwargss M/var/www/html/auto_sub_bot/venv/lib/python3.11/site-packages/torch/storage.py__init__z_StorageBase.__init__returncdSrrrs r __len__z_StorageBase.__len__r"r#cdSrrridxs r __getitem__z_StorageBase.__getitem__r"r#cdSrrrs r __setitem__z_StorageBase.__setitem__ r"r#Nsource non_blockingcdSrrrr.r/s r copy_z_StorageBase.copy_!r"r#cdSrrr&s r newz_StorageBase.new"r"r#cdSrrr&s r nbytesz_StorageBase.nbytes#r"r#c*|Sr)r6r&s r sizez_StorageBase.size%s{{}}r#dtypecdSrr)rr9r/s r typez_StorageBase.type(r"r#c dSrrrrr/rs r cudaz_StorageBase.cuda)r"r#c dSrrr=s r hpuz_StorageBase.hpu*r"r#cdSrrr&s r element_sizez_StorageBase.element_size+r"r#c|jjSr)rindexr&s r get_devicez_StorageBase.get_device-s {  r#cdSrrr&s r data_ptrz_StorageBase.data_ptr0r"r#cdSrrrs r _share_filename_cpu_z!_StorageBase._share_filename_cpu_3r"r#cdSrrrs r _share_fd_cpu_z_StorageBase._share_fd_cpu_4r"r#clsr8cdSrrrLr8s r _new_using_filename_cpuz$_StorageBase._new_using_filename_cpu5s?Bsr#cdSrrrNs r _new_using_fd_cpuz_StorageBase._new_using_fd_cpu79<r#cdSrrrLrrs r from_bufferz_StorageBase.from_buffer9rRr#)rr9cdSrr)rLmanagerobjr8rr9s r _new_shared_filename_cpuz%_StorageBase._new_shared_filename_cpu;sehehr#cdSrrrTs r _release_ipc_counter_cudaz&_StorageBase._release_ipc_counter_cuda=sGJsr#cdSrrrTs r _new_with_weak_ptrz_StorageBase._new_with_weak_ptr?s@Cr#cdSrrr&s r _shared_decrefz_StorageBase._shared_decrefAr"r#cdSrrrs r _write_filez_StorageBase._write_fileBr"r#cdSrrrr8s r resize_z_StorageBase.resize_Cr"r#cdSrrrs r _weak_refz_StorageBase._weak_refDr"r#cdSrrrs r _set_from_filez_StorageBase._set_from_fileEr"r#cdSrrrs r _set_cdataz_StorageBase._set_cdataFr"r#cdSrrrs r _share_cuda_z_StorageBase._share_cuda_Gr"r#cdSrrr&s r is_sharedz_StorageBase.is_sharedHr"r#cdSrrrTs r _new_shared_cudaz_StorageBase._new_shared_cudaIs>Acr#cdSrrrs r _shared_increfz_StorageBase._shared_increfKr"r#cdSrrrTs r _free_weak_refz_StorageBase._free_weak_refLs.1cr#cdSrrr&s r is_cudaz_StorageBase.is_cudaNssr#cdSrrr&s r is_hpuz_StorageBase.is_hpuPscr#cdSrr)rLfilenamesharedr6s r from_filez_StorageBase.from_fileRs7:sr#cdSrrrTs r _expiredz_StorageBase._expiredTs-0Sr#cdSrrrs r _byteswapz_StorageBase._byteswapVr"r#cdSrrrs r _get_filenamez_StorageBase._get_filenameWr"r#c$dtjdjdtd}jjdkrd|zSddfd t Dz}|d z|zS) N[z(device= ) of size ]meta...   c3BK|]}t|VdSrstr.0irs r z'_StorageBase.__str__..`-'Q'QDG 'Q'Q'Q'Q'Q'Qr# )torchtypenamerlenr;joinranger8rinfo_strdata_strs` r __str__z_StorageBase.__str__Ys $t$$ $ $dk $ $4yy $ $ $  ; v % %X% %UZZ'Q'Q'Q'QeDIIKK>P>P'Q'Q'QQQQHd?X- -r#c t|Srrr&s r __repr__z_StorageBase.__repr__cs4yyr#cxtfdtDS)Nc3(K|] }|V dSrrrs r rz(_StorageBase.__iter__..g'88DG888888r#)iterrr8r&s`r __iter__z_StorageBase.__iter__fs68888U499;;%7%7888888r#c*|Sr)cloner&s r __copy__z_StorageBase.__copy__iszz||r#c|di}|j|vr ||jS|}|||j<|S)Nr) setdefaultrr)rmemo new_storages r __deepcopy__z_StorageBase.__deepcopy__lsNw++ ;$   $ $jjll 'T[r#ctj}tj||dt|ffSNF)_use_new_zipfile_serializationioBytesIOrsave_load_from_bytesgetvaluerbs r __reduce__z_StorageBase.__reduce__t9 JLL 45AAAA 1::<ctjgtj|jt t ||S)zDetermine whether the CPU storage is already pinned on device. Args: device (str or torch.device): The device to pin memory on. Default: ``'cuda'``. Returns: A boolean variable. r)rrrrrrr is_pinnedrrs r r z_StorageBase.is_pinnedsK|Bek$+FFFKK $  !!!*6!2!2 3r#cR|jjdkr%td|dtjgtj|jtt| |}| S)zCopy the CPU storage to pinned memory, if it's not already pinned. Args: device (str or torch.device): The device to pin memory on. Default: ``'cuda'``. Returns: A pinned CPU storage. rz cannot pin 'z' only CPU memory can be pinnedr) rr;rrrrrrr pin_memoryuntyped_storage)rr pinned_tensors r r z_StorageBase.pin_memorys ; u $ $W499;;WWWXX X Ru{4;OOOTT $  !!!+F!3!3 ,,...r#cddlm}|jjdtjfvrn7|dkr|n||S).See :meth:`torch.UntypedStorage.share_memory_`rget_sharing_strategyr> file_system) torch.multiprocessingrrr;r_C_get_privateuse1_backend_namerIrK)rrs r share_memory_z_StorageBase.share_memory_sz>>>>>> ; (N(N(P(PQ Q Q  ! ! # #} 4 4  % % ' ' ' '    ! ! ! r#rrcddlm}tj|}|jdtjfvr |||S|dkr||S||S)>Create a new storage in shared memory with the same data type.rrr>rr) rrrrr;rrrOrQ)rLr8rrs r _new_sharedz_StorageBase._new_shareds ?>>>>>f%% ;658#I#I#K#KL L L3tF+++ + ! ! # #} 4 4..t44 4((.. .r#c|Srrr&s r untypedz_StorageBase.untypeds r#ctj|}|jr t t |dz d}||dS)zSwap bytes in underlying data.rN)r_utils _element_size is_complexmaxrr)rr9 elem_sizes r byteswapz_StorageBase.byteswapsUL..u55   3C A ..22I y!!!!!r#rNFr>)X__name__ __module__ __qualname__r __annotations__rrrrrr!rr'r+r-r _Optionalr2r4r6r8rr;r>r@rBrErGrIrK classmethodr rOrQrUrYr[r]r_rardrfrhrjrlrnrprrrtpropertyrvrxr|r~rrrrrrrrrrrrrrrrrrrrrrrrrrrr r rrrr% __classcell__rs@r rrs KKKItM4 L,,,!!!!!###///NNANYt_NNNNNQ     cWV)C.VtVPQVVVVGGGGGGFFFFFF&c&&&&!C!!!!##""""988222BT!WBCBABBB[B.wrappers   +C'''+C0)2):):!#&!#&..000                                   32d,T,,,V,,"{g----'33%g.66888)'23333333333333333 #w"{g----'33%g.66888)'23333333333333333 #sfAA44A8;A8BBB C?>(C22C6 9C6 ?E(E > E E EE E) functoolswraps)r;r<s` r _share_memory_lock_protectedr?s8_R33333< Nr#ceZdZfdZedZedZedeefdZ e fdZ e fdZ e fdZ xZS) rct|jjdkrtdtj|i|S)Nr$Not available for 'meta' device type)rr;NotImplementedErrorrr+rrrrs r r+zUntypedStorage.__getitem__4s> ; v % %%&LMM M"uww"D3F333r#c"|jjdkSNr>rr;r&s r rvzUntypedStorage.is_cuda9s{6))r#c"|jjdkSNr@rGr&s r rxzUntypedStorage.is_hpu=s{5((r#r$c*|SzReturns the file name associated with this storage if the storage was memory mapped from a file. or ``None`` if the storage was not created by memory mapping a file.)rr&s r rzzUntypedStorage.filenameAs!!###r#c6tj|i|S)a{ Moves the storage to shared memory. This is a no-op for storages already in shared memory and for CUDA storages, which do not need to be moved for sharing across processes. Storages in shared memory cannot be resized. Note that to mitigate issues like `this `_ it is thread safe to call this function from multiple threads on the same object. It is NOT thread safe though to call any other function on self without proper synchronization. Please see :doc:`/notes/multiprocessing` for more details. .. note:: When all references to a storage in shared memory are deleted, the associated shared memory object will also be deleted. PyTorch has a special cleanup process to ensure that this happens even if the current process exits unexpectedly. It is worth noting the difference between :meth:`share_memory_` and :meth:`from_file` with ``shared = True`` #. ``share_memory_`` uses `shm_open(3) `_ to create a POSIX shared memory object while :meth:`from_file` uses `open(2) `_ to open the filename passed by the user. #. Both use an `mmap(2) call `_ with ``MAP_SHARED`` to map the file/object into the current virtual address space #. ``share_memory_`` will call ``shm_unlink(3)`` on the object after mapping it to make sure the shared memory object is freed when no process has the object open. ``torch.from_file(shared=True)`` does not unlink the file. This file is persistent and will remain until it is deleted by the user. Returns: ``self`` )rrrDs r rzUntypedStorage.share_memory_Gs"B%uww$d5f555r#c6tj|i|Sr)rrKrDs r rKzUntypedStorage._share_fd_cpu_js%uww%t6v666r#c6tj|i|Sr)rrIrDs r rIz#UntypedStorage._share_filename_cpu_ns+uww+Tz._storage_type_to_dtype_map..s1GGGS#SGGGr#)rjitems) dtype_maps r _storage_type_to_dtype_maprqs=GG!;!=!=!C!C!E!EGGGI r#c |tjtjtjtjtjfvrtjtjtjtjtjtjtjtjtjtji}tj ||||}ntj |||}| j S)Nr) rrgrhrirfrerint32rrr_untyped_storage)sequencer9rinterpret_dtypes tmp_tensors r _get_storage_from_sequencerxs u~u~u|U[YYY L%+ NEK NEK L%+ K  \ "5) \   $ $ & & 77r#c|tr!t|ttjfSt|tSr) HAS_NUMPYrrnpinteger)xs r _isintr~s1"!c2:.///!S!!!r#FctSr)"_always_warn_typed_storage_removalrr#r &_get_always_warn_typed_storage_removalrs --r#c8t|tsJ|adSr)rrr) always_warns r &_set_always_warn_typed_storage_removalrs$ k4 ( ((( ()4&&&r#rcd}ts |r2d}tj|t|dzdtjd<dSdS)NcVttdsdStjd S)N has_warnedT)hasattr_warn_typed_storage_removal__dict__rr#r is_first_timez2_warn_typed_storage_removal..is_first_times,2LAA J42;LII Ir#aTypedStorage is deprecated. It will be removed in the future and UntypedStorage will be the only storage class. This should only matter to you if you are using storages directly. To access UntypedStorage directly, use tensor.untyped_storage() instead of tensor.storage()r) stacklevelTr)rwarningswarn UserWarningrr)rrmessages r rrs{JJJ .//B==??B Q   g{zA~FFFF=A#,\:::BBr#c$dtjd<dS)NFr)rrrr#r !_reset_warn_typed_storage_removalrs9>(666r#modulec|dddtjfvr|ddSdS)N.r>r)splitrrr)rs r _get_device_from_modulersM ||C)O)O)Q)Q RRR||C  $$ur#c eZdZUdZejed<edZede e fdZ dZ dddddfd Z ddddd d Zed Zed ZdZdZdZdadZdZdZdZdZdbdede efdZdZdZdcde e dedeee ffdZdcdefdZ dcdefdZ!dZ"dZ#de$fd Z%d!Z&d"Z'd#Z(d$Z)d%Z*d&Z+fd'Z,d(Z-d)Z.d*Z/ddd,ee ej0ffd-Z1ddd,ee ej0ffd.Z2d/Z3d0Z4dd1d2Z5ed3Z6ed4Z0d5Z7d6Z8d7Z9d8Z:d9Z;d:Zd=Z?e@d>ZAd?ZBe@d@ZCe@dddAdBZDdCZEdDZFdEZGdFZHdGZIdHZ$dIZJdJZKdKZLdLZdMZMdNZNdOZOdPZPdQZQe@dRZRe@dSZSdTZTdUZUdVZVdWZWdXZXdYZYe@dZZZd[Z[d\Z\e@dd1d]Z]d^Z^d_Z_d`Z`xZaS)e TypedStorageFr9c|jSr)r9r&s r _dtypezTypedStorage._dtypes zr#r$c|jjSrK)rtrzr&s r rzzTypedStorage.filenames$--r#ct|td|||S)Nr)r_setitemslice_size)rvalues r fill_zTypedStorage.fill_s9#%%% eAtzz||,,e444 r#N) wrap_storager9r _internalc|st|tjjkrt d|t kr!t |S|d}|t |dz|t |dz|t|dkrt |dzt|dkrbt|dsMt|dtj j s(t|dt|dzt ||jt#|jd d St|dkrt |d zt|tjs"t|d t|zt#|j}|jj|kr"t |d |d|jjzt |||jd dS)Nz8Only child classes of _LegacyStorage can be instantiatedz.__new__ received an invalid combination of arguments. Expected one of: * no arguments * (int size) * (Sequence data) * (*, UntypedStorage wrap_storage)z. Keyword argument 'device' cannot be specifiedz- Keyword argument 'dtype' cannot be specifiedr Too many positional argumentsr Argument type not recognized: T)r9rrB No positional arguments should be given when using 'wrap_storage'9 Argument 'wrap_storage' must be UntypedStorage, but got z" Device of 'wrap_storage' must be z , but got rr9r)rrr_LegacyStorage RuntimeErrorrr__new__rr~r collectionsabcSequencerr;rrr)rrr9) rLrr9rrr arg_error_msg cls_devicers r rzTypedStorage.__new__s * ' ) ) ) %-. . .YZZ Z ,  77??3'' '666 !"!EFGGG "!DEFFF#t99q==&%9:;;;t99>>&a//>*TRSWVaVeVnBoBo>#%J4Q==JJKLLL$*23>BB" $$$$t99>>&%))*** ",0DEEk#%iUYZfUgUgiijkkk5S^DD &+z99&%@j@@%1%8%=@@@AAA $!-)" $$$$r#)rr9rrc,|std}|t|dkrt|dz|t|dzt|tjs"t |dt|z|t|dz||_t|tjs"t |dt|z||_ dS|t j n||_t j |dn|}|jtj tj tjtjtjfvr|jd krtd t|dkrt j| |_ dSt|d krt#|drEt jt%|d|z| |_ dSt|dt(jjr#t/|d|j||_ dSt |d t|dzt|dz)Na1TypedStorage.__init__ received an invalid combination of arguments. Expected one of: * (*, torch.device device, torch.dtype dtype) * (int size, *, torch.device device, torch.dtype dtype) * (Sequence data, *, torch.device device, torch.dtype dtype) * (*, UntypedStorage wrap_storage, torch.dtype dtype)rrz# Argument 'dtype' must be specifiedz+ Argument 'dtype' must be torch.dtype, not zG Argument 'device' should not be specified when 'wrap_storage' is givenrrr>/Cannot create CUDA storage with quantized dtyperrrr)rrrrrr9rr;rrtget_default_dtyperrgrhrirfrer~rr!rrrrx)rrr9rrrrs r r!zTypedStorage.__init__:s * ' ) ) ) E   #4yyA~~"!%%&&& }"!:;<<<eU[11 R!P4;;PPQRRR!"!^_```DJlE,@AA g!eQUVbQcQceefggg%1D ! ! !7%%vFFFzelENENELZ_Zefff;&((&'XYYY4yyA~~(-(  ! ' '(? N N N N  ! ' ' = = = r#cFt|Sr)r_nbytesr&s r r6zTypedStorage.nbytess#%%%||~~r#c4|jSr)rtr6r&s r rzTypedStorage._nbytess$++---r#ct|V|}||jdz|jzSd|jt |jgS|j||S)Nr)r_get_legacy_storage_classr)r(rr;rt)rr9r/ legacy_classs r r;zTypedStorage.types}#%%% =99;;L'#.4|7LLL88T_d4jj.ABCC C(--e\BB Br#c t|jtjtjtjtjtjfvrtd|j j ||fi|}| |S)Nr) rr9rrgrhrirfrerrtr>r)rrr/r cuda_storages r r>zTypedStorage.cudasv#%%% :%, V[Vab b bPQQ Q-GT-B-GP\-g-g`f-g-g ((666r#c t|jtjtjtjtjtjfvrtd|j j ||fi|}| |S)Nz.Cannot create HPU storage with quantized dtype) rr9rrgrhrirfrerrtr@r)rrr/r hpu_storages r r@zTypedStorage.hpusu#%%% :%, V[Vab b bOPP P,ED,A,Efl,e,e^d,e,e ((555r#cFt|Sr)rr!r&s r rBzTypedStorage.element_sizes #%%%!!###r#cJtj|jSr)rr r!r9r&s r r!zTypedStorage._element_size#s|))$*555r#cPt|jSr)rrtrEr&s r rEzTypedStorage.get_device&s##%%%$//111r#c Ptdtjdjdjdt d }jjdkrd|zSdd fd t Dz}|d z|zS) Nrz(dtype=z , device=rrrrrrc3BK|]}t|VdSrrrs r rz'TypedStorage.__str__..2rr#r) rrrr9rrr;rrr8rs` r rzTypedStorage.__str__*s#%%% :t$$ : :TZ : :k : :-0YY : : :  ; v % %X% %UZZ'Q'Q'Q'QeDIIKK>P>P'Q'Q'QQQQHd?X- -r#c<tt|Sr)rrr&s r rzTypedStorage.__repr__5s#%%%4yyr#cttfdtDS)Nc3(K|] }|V dSrrrs r rz(TypedStorage.__iter__..;rr#)rrrr8r&s`r rzTypedStorage.__iter__9sB#%%%8888U499;;%7%7888888r#cvt|tj|jSr)rrcopyrtr&s r rzTypedStorage.__copy__=s/#%%%((43H)I)IJJJr#cHt||Sr)r _deepcopyrrs r rzTypedStorage.__deepcopy__As #%%%~~d###r#c\|tj|j|Sr)rrdeepcopyrtrs r rzTypedStorage._deepcopyFs%((t7Ld)S)STTTr#ctt|zSr)rrrr6rs r rzTypedStorage.__sizeof__Is2#%%%ww!!##dkkmm33r#cvt||jS)r)rrrtrr&s r rzTypedStorage.cloneMs1#%%%(()>)D)D)F)FGGGr#c<tt|Sr)rrr&s r rzTypedStorage.tolistRs#%%%Dzzr#cvt||jS)r)rrrtrr&s r rzTypedStorage.cpuWs1#%%%(()>)B)B)D)DEEEr#r>rcRt|j|S)zDetermine whether the CPU TypedStorage is already pinned on device. Args: device (str or torch.device): The device to pin memory on. Default: ``'cuda'`` Returns: A boolean variable. )rrtr r s r r zTypedStorage.is_pinned\s' $%%%$..v666r#czt||j|S)zCopy the CPU TypedStorage to pinned memory, if it's not already pinned. Args: device (str or torch.device): The device to pin memory on. Default: ``'cuda'``. Returns: A pinned CPU storage. r)rrrtr r s r r zTypedStorage.pin_memoryhs9 $%%%(()>)I)IQW)I)X)XYYYr#cFt|S)r)r_share_memory_r&s r rzTypedStorage.share_memory_ts #%%%""$$$r#c8|j|Sr)rtrr&s r rzTypedStorage._share_memory_zs ++--- r#rc|d}tj|}tj||z|}t ||jdS)rNrrTr)rrrrr!rr9)rr8rrs r rzTypedStorage._new_shared~sj >Ff%%.::4$BTBTBVBV;V_e:ff(* r#c|jjSr)rtrr&s r rzTypedStorage._cdatas$++r#c6t|jjSr)rrtrr&s r rzTypedStorage.devices#%%%$++r#cFt|Srrr&s r r8zTypedStorage.sizerr#c^|j|zSr)rtr6r!r&s r rzTypedStorage._sizes*$++--1C1C1E1EEEr#cFt|Sr)r_pickle_storage_typer&s r pickle_storage_typez TypedStorage.pickle_storage_types #%%%((***r#c t|jS#t$r}td|jd|d}~wwxYw)Nzdtype z is not recognized)rjr9KeyError)res r rz!TypedStorage._pickle_storage_types\ K-// ; ; K K KBDJBBBCC J Ks A>Actj}tj||dt|ffSrrrs r rzTypedStorage.__reduce__rr#cFt|Sr)r _data_ptrr&s r rGzTypedStorage.data_ptrs#%%%~~r#c4|jSr)rtrGr&s r rzTypedStorage._data_ptrs$--///r#cLt||dSr)r_resize_rcs r rdzTypedStorage.resize_s&#%%% dr#cd|j||zdSr)rtrdr!rcs r rzTypedStorage._resize_s0 %%dT-?-?-A-A&ABBBBBr#c$tj|i|Sr)rrtrTs r rtzTypedStorage._free_weak_refs,d=f===r#c&|jj|i|Sr)rtrfrs r rfzTypedStorage._weak_refs.t$.????r#c8t|j|i|Sr)r _from_bufferrTs r rUzTypedStorage.from_buffers&#%%%s0000r#rc|tkrj|tjn|}tj|dn|}|jdkrt d|jtjj|d|i|}nT|t|dkrt d|t d|j }tjj|d|i|}t||dS) Nrz3TypedStorage.from_buffer: Not available for device r9zefrom_buffer: 'dtype' can only be specified in UntypedStorage.from_buffer and TypedStorage.from_bufferzffrom_buffer: 'device' can only be specified in UntypedStorage.from_buffer and TypedStorage.from_bufferTr) rrrrr;rrrUrr)rLr9rrrrs r rzTypedStorage._from_buffers  ,  16E+---EE\6>%%vFFF{e##"#fY_Yd#f#fggg494H4TVZ4rbg4rkq4r4rOO CIINN"NOOO!"NOOOJE#2>\E\U[\\O( r#ct|tjstdt |tjg|j|j|| }| | kr| }|Sr) rrr9rr;rrrrrrGrrs r rzTypedStorage._tos%-- XVeVVWW W,rDKHHHMMdSSVVW\]]llnn      0 0mmooGr#c\t|tjSr)rrrrr&s r rzTypedStorage.double"#%%%xx %%%r#c\t|tjSr)rrrrr&s r rzTypedStorage.float"#%%%xx $$$r#c\t|tjSr)rrrrr&s r rzTypedStorage.half"#%%%xx ###r#c\t|tjSr)rrrrr&s r rzTypedStorage.longrr#c\t|tjSr)rrrrr&s r rzTypedStorage.ints"#%%%xx """r#c\t|tjSr)rrrrr&s r rzTypedStorage.shortrr#c\t|tjSr)rrrrr&s r rzTypedStorage.char rr#c\t|tjSr)rrrrr&s r rzTypedStorage.byterr#c\t|tjSr)rrrrr&s r rzTypedStorage.boolrr#c\t|tjSr)rrrrr&s r rzTypedStorage.bfloat16s"#%%%xx'''r#c\t|tjSr)rrrrr&s r rzTypedStorage.complex_doubles"#%%%xx &&&r#c\t|tjSr)rrrrr&s r rzTypedStorage.complex_float"r r#c\t|tjSr)rrrrr&s r rzTypedStorage.float8_e5m2's##%%%xx)***r#c\t|tjSr)rrrrr&s r rzTypedStorage.float8_e4m3fn,s##%%%xx+,,,r#ct|tkrtdt|||t j|jz}||}|S)afrom_file(filename, shared=False, size=0) -> Storage Creates a CPU storage backed by a memory-mapped file. If ``shared`` is ``True``, then memory is shared between all processes. All changes are written to the file. If ``shared`` is ``False``, then the changes on the storage do not affect the file. ``size`` is the number of elements in the storage. If ``shared`` is ``False``, then the file must contain at least ``size * sizeof(Type)`` bytes (``Type`` is the type of storage). If ``shared`` is ``True`` the file will be created if needed. Args: filename (str): file name to map shared (bool): whether to share memory (whether ``MAP_SHARED`` or ``MAP_PRIVATE`` is passed to the underlying `mmap(2) call `_) size (int): number of elements in the storage z/from_file can only be called on derived classesr) rrrrr|rr r!r9)rLrzr{r8rrs r r|zTypedStorage.from_file1sx( $%%% ,  PQQ Q*8*B*B   5<--ci88 8+:+:#?333r#c$tj|i|Sr)rr~rTs r r~zTypedStorage._expiredOs&7777r#c&|jj|i|Sr)rtrars r razTypedStorage._write_fileSs0t$0$A&AAAr#c&|jj|i|Sr)rtrhrs r rhzTypedStorage._set_from_fileV3t$3TDVDDDr#c&|jj|i|Sr)rtrjrs r rjzTypedStorage._set_cdataYs/t$/@@@@r#c&|jj|i|Sr)rtrlrs r rlzTypedStorage._share_cuda_\s1t$14B6BBBr#cFt|Sr)r _is_sharedr&s r rnzTypedStorage.is_shared_s#%%%   r#c4|jSr)rtrnr&s r r%zTypedStorage._is_sharedds$..000r#c.tjj|i|Sr)rrrprTs r rpzTypedStorage._new_shared_cudags#4dEfEEEr#cb|jj|i|\}}}||||zfSr)rtrIr!)rrrmanager_handlestorage_handler8s r rIz!TypedStorage._share_filename_cpu_ksC/Yt/D/Y[_/jci/j/j,~tt7I7I7K7K/KKKr#c8|j|Sr)rtr_r&s r r_zTypedStorage._shared_decrefos ,,... r#c.tjj|i|Srrrr[)rLrrrs r _release_ipc_counterz!TypedStorage._release_ipc_counters#=tNvNNNr#c&|jj|i|Sr)rtrrrs r rrzTypedStorage._shared_increfwr!r#c^|jj|i|\}}|||zfSr)rtrKr!)rrrfdr8s r rKzTypedStorage._share_fd_cpu_zs<74(7HHHD44--/////r#cp|jtvrdSt|j}|jjddtjfvrdS|jjdkrtntt|jj} t||S#t$rYdSwxYw)Nrr>) r9rjrr;rrrgetattrAttributeError)r storage_namers r rz&TypedStorage._get_legacy_storage_class~s :799 9 94133DJ? ; E6583Y3Y3[3[#\ \ \4+*e33 HX9Y9Y 6<00 0   44 sB'' B54B5)Frr&r')br(r)r*rrr9r+r.rr,rrzrrr!rvrxrrr'rr-rr+rrrr2r6rrr;r>r@rBr!rrErrrrrrrrrrrr r rrrrr8rrrrrGrrdrr-rtrfrUrrrrrrrrrrrrrrr|r~rarhrjrlrnr%rprIr_r.rrrKrr/r0s@r rrs<I ; X.)C....X.  *.T$RWF$F$F$F$F$F$F$P&*DTYB7B7B7B7B7H;;X;::X:%%% < < <****2)))   0"""...<AYt_ ... C C)C. Ct CPUVWY\V\P] C C C C777777666666$$$ 6662C2222 . . .999KKK$$$ UUU44444HHH  FFF 7 7c5<&7 8 7 7 7 7 Z ZsEL'8!9 Z Z Z Z%%% +/     ,,X,,,X, FFF +++ KKK 333    000 CCC>>[>@@@11[1'+D[4&&& %%% $$$ $$$ ### %%% $$$ %%% $$$ ((( ''' &&& +++ --- [:88[8BBBEEEAAACCC!!! 111FF[FLLL04OOOO[OEEE000r#rc*eZdZUejed<dZdS)_LegacyStorageMetar9ct|tkr4t|j}||jjko|j|jkSdSr&)r;rrr)rr9)rLinstancers r __instancecheck__z$_LegacyStorageMeta.__instancecheck__sE >>\ ) )0@@J(/"66YSY(.=X Yur#N)r(r)r*rr9r+r;rr#r r8r8s3 ;r#r8cPeZdZedZedZedZdS)rctj||z}||S)rr)rrrr!)rLr8rs r rz_LegacyStorage._new_sharedsD .::4##%%BUBUBWBW;WXXs0000r#c.tjj|i|Srr-rTs r r.z#_LegacyStorage._release_ipc_counterr/r#c|tj|jz}|tj|||S)Nr)rr r!r9rrY)rLrWrXr8 bytes_sizes r _new_shared_filenamez#_LegacyStorage._new_shared_filenamesHEL66syAAA s 4 M MgWZ\f g ghhhhr#N)r(r)r*r-rr.rArr#r rrsk11[1 OO[Oii[iiir#r) metaclassrcx t|S#t$r}td|d|d}~wwxYw)Nzpickle storage type "z" is not recognized)rqr)rrs r #_get_dtype_from_pickle_storage_typerDsdU)++,?@@ UUU L$7 L L LNNST UUs 949)r)r@rjrqrxr~rrrrrrrr__doc__r8rrDrr#r rMs &&&&&&&&&&XXXXXXXXXXXXXXXXXX II BBB$Y^%%135io-.333 GC:;;;w"w"w"w"w"w"w"w"t   D==========UX)<======~%%%    44 4 888,""" &+"...555 BBBB&???C n n n n n n n n `"M !M < iiiii\-?iiii USUUUUUUsAAA