tj3UddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z ddl mZddlZddlmZmZddlmZmZmZe rddlmZdd gZ ddlZd Zn#e$rd ZdZYnwxYwejZiZ d e!d <e ddZ"GddZ#dZ$Gdd ej%j&e#Z'dZ(ej)dZ*ej)dZ+ej)dZ,dZ-dZ.d a/dZ0dZ1d+dZ2dZ3d,d"Z4Gd#dZ5ej6e5j7_6e#j8j6e5j8_6e#j9j6e5j9_6ej6e5j:_6Gd$d%e7Z;Gd&d'e5e;(ZZ-e,d?Z.edmd@Z/edmdAZ0dBZ1dtdCZ2dDZ3dEZ4dFZ5dGZ6dHZ7fdIZ8dJZ9dKZ:dLZ;dMZdRZ?dSZ@dTZAdUZBdVZCdWZDdXZEdYZFdZZGd[ZHd\ZId]ZJd^ZKd_ZLd`ZMdaZNdkdvdcZOdkdvddZPdeZQedfd/dgZRdhZSdiZTxZUS)w _StorageBaser_cdataFr is_sparse is_sparse_csrz torch.devicedeviceNtorch.device | None _fake_devicez int | None_checkpoint_offsetcdSNselfargskwargss K/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/storage.py__init__z_StorageBase.__init__5s returnr ctrNotImplementedErrorr"s r%__len__z_StorageBase.__len__8!!r'ctrr*r"idxs r% __getitem__z_StorageBase.__getitem__;r.r'ctrr*r!s r% __setitem__z_StorageBase.__setitem__>r.r'sourcer non_blocking _bool | Nonectrr*r"r5r6s r%copy_z_StorageBase.copy_Ar.r'rctrr*r,s r%newz_StorageBase.newDr.r'ctrr*r,s r%nbytesz_StorageBase.nbytesGr.r'c*|Sr)r>r,s r%sizez_StorageBase.sizeJs{{}}r'dtype str | Nonec$t|||Sr)r )r"rAr6s r%typez_StorageBase.typeMsT5,///r'c|rtjd|ntjd}|||S)aReturns a copy of this object in CUDA memory. If this object is already in CUDA memory and on the correct device, then no copy is performed and the original object is returned. Args: device (int): The destination GPU id. Defaults to the current device. non_blocking (bool): If ``True`` and the source is in pinned memory, the copy will be asynchronous with respect to the host. Otherwise, the argument has no effect. cudarr6torchrtor"rr6device2s r%rFz_StorageBase.cudaRs?39R%,vv...el6>R>RwwgLwAAAr'c|rtjd|ntjd}|||S)aReturns a copy of this object in HPU memory. If this object is already in HPU memory and on the correct device, then no copy is performed and the original object is returned. Args: device (int): The destination HPU id. Defaults to the current device. non_blocking (bool): If ``True`` and the source is in pinned memory, the copy will be asynchronous with respect to the host. Otherwise, the argument has no effect. hpurGrHrKs r%rNz_StorageBase.hpuas?28P%,uf---U\%=P=PwwgLwAAAr'ctrr*r,s r% element_sizez_StorageBase.element_sizepr.r'c|jjSr)rindexr,s r% get_devicez_StorageBase.get_devicess {  r'ctrr*r,s r%data_ptrz_StorageBase.data_ptrvr.r'ctrr*r,s r% resizablez_StorageBase.resizableyr.r'ctrr*r!s r%_share_filename_cpu_z!_StorageBase._share_filename_cpu_}r.r'ctrr*r!s r%_share_fd_cpu_z_StorageBase._share_fd_cpu_r.r'r@rctrr*clsr@s r%_new_using_filename_cpuz$_StorageBase._new_using_filename_cpu!!r'ctrr*r]s r%_new_using_fd_cpuz_StorageBase._new_using_fd_cpur`r'ctrr*r^r#r$s r% from_bufferz_StorageBase.from_bufferr`r')rrActrr*)r^managerobjr@rrAs r%_new_shared_filename_cpuz%_StorageBase._new_shared_filename_cpus "!r'rc|j|i|Sr)_release_ipc_counter_cudar^rr#r$s r%_release_ipc_counterz!_StorageBase._release_ipc_counters,s,d=f===r'ctrr*rds r%rlz&_StorageBase._release_ipc_counter_cudar`r'ctrr*rds r%_new_with_weak_ptrz_StorageBase._new_with_weak_ptrr`r'ctrr*r,s r%_shared_decrefz_StorageBase._shared_decrefr.r'ctrr*r!s r% _write_filez_StorageBase._write_filer.r'ctrr*r"r@s r%resize_z_StorageBase.resize_r.r'ctrr*r!s r% _weak_refz_StorageBase._weak_refr.r'ctrr*r!s r%_set_from_filez_StorageBase._set_from_filer.r'ctrr*r!s r% _set_cdataz_StorageBase._set_cdatar.r'ctrr*r!s r% _share_cuda_z_StorageBase._share_cuda_r.r'ctrr*r,s r% is_sharedz_StorageBase.is_sharedr.r'ctrr*rds r%_new_shared_cudaz_StorageBase._new_shared_cudar`r'ctrr*r!s r%_shared_increfz_StorageBase._shared_increfr.r'ctrr*rds r%_free_weak_refz_StorageBase._free_weak_refr`r'ctrr*r,s r%is_cudaz_StorageBase.is_cudar`r'ctrr*r,s r%is_hpuz_StorageBase.is_hpur`r'ctrr*)r^filenamesharedr>s r% from_filez_StorageBase.from_filer`r'ctrr*rds r%_expiredz_StorageBase._expiredr`r'ctrr*r!s r% _byteswapz_StorageBase._byteswapr.r'ctrr*r!s r% _get_filenamez_StorageBase._get_filenamer.r'c$dtjdjdtd}jjdkrd|zSddfd t Dz}|d z|zS) N[z(device= ) of size ]meta...   c3BK|]}t|VdSrstr.0ir"s r% z(_StorageBase.__repr__..s-#M#MQCQLL#M#M#M#M#M#Mr' )rItypenamerlenrDjoinranger@r"info_strdata_strs` r%__repr__z_StorageBase.__repr__sXu~d++XXT[XXCPTIIXXX ; v % %X% %#M#M#M#M% :L:L#M#M#MMMM$))r'cxtfdtDS)Nc3(K|] }|V dSrr rs r%rz(_StorageBase.__iter__..'88DG888888r')iterrr@r,s`r%__iter__z_StorageBase.__iter__s68888U499;;%7%7888888r'c*|Sr)cloner,s r%__copy__z_StorageBase.__copy__szz||r'c|di}|j|vr ||jS|}|||j<|S)NrI) setdefaultrr)r"memo new_storages r% __deepcopy__z_StorageBase.__deepcopy__sNw++ ;$   $ $jjll 'T[r'ctj}tj||dt|ffSNF)_use_new_zipfile_serializationioBytesIOrIsave_load_from_bytesgetvaluer"bs r% __reduce__z_StorageBase.__reduce__9 JLL 45AAAA 1::<rr:r,s r%rz_StorageBase.clones5tDzz$++-- <<<BB4HHHr'c t|Sz6Return a list containing the elements of this storage.)listr,s r%tolistz_StorageBase.tolistsDzzr'c|jjdkr:tj||dS|S)AReturn a CPU copy of this storage if it's not already on the CPU.cpuFrrDrIrr@r:r,s r%rz_StorageBase.cpus@ ; u $ $' 44::4GG G r'c|jjdkr>>>>> ; ? 2 2  ! ! # #} 4 4  % % ' ' ' '    ! ! ! r'rcddlm}tj|}|jdkr |||S|dkr||S||S)>Create a new storage in shared memory with the same data type.rr(rrjr*)r+r)rIrrDr_rb)r^r@rr)s r% _new_sharedz_StorageBase._new_shareds ?>>>>>f%% ;%  3tF+++ + ! ! # #} 4 4..t44 4((.. .r'c|Srr r,s r%untypedz_StorageBase.untypeds r'ctj|}|jr t t |dz d}||dS)zSwap bytes in underlying data.N)rI_utils _element_size is_complexmaxrr)r"rA elem_sizes r%byteswapz_StorageBase.byteswapsUL..u55   3C A ..22I y!!!!!r'r(r r)r5rr6r7r(r)r(rNF)rArBr6r r(r)r(r )r@r r(rr(r)r@r r(rB)rrr6r rr)V__name__ __module__ __qualname____annotations__rrrrr&r-r2r4r:r<r>r@rDrFrNrPrSrUrWrYr[ classmethodr_rbrerirnrlrqrsrurxrzr|r~rrrrrpropertyrrrrrrrrrrrrrrrrr rJrrrrrrrrrrr rrrrrr r#r,r/r1r: __classcell__rs@r%rr)sKKKI M    )-L,,,,%)))))   """""""""""""""""""""""?D00000 B B B B B B B B B B""""!!!!"""""""""""""""""[""""[""""[" " " " " "[ "04>>>>[>"""[""""[""""""""""""""""""""""""""""""""[""""""["""X"""X""""[""""[""""""""***999333 22222III    CH////// &&&%%%$$$$$$###%%%$$$%%%$$$((('''&&&+++---//////     /////(   ). / / / /[ /"""""""r'rcFtjfd}|S)Ncd}d}t5|j}|tvrt|}n.wrappers   +C'''+C0)2):):!#&!#&..000                                   32d,T,,,V,,";'))(dddW[Wbdd(33%g.66888)'23333333333333333#w";'))(dddW[Wbdd(33%g.66888)'23333333333333333#sfAA44A8;A8BBB D(D  D D /F(E:. F:E> >FE> F) functoolswraps)rSrTs` r%_share_memory_lock_protectedrWs9_R 3 3 3 3 3D Nr'ceZdZfdZedZedZed dZefdZ efdZ efd Z xZ S) rct|jjdkrtdtj|i|S)Nr$Not available for 'meta' device type)rrDr+rr2r"r#r$rs r%r2zUntypedStorage.__getitem__s> ; v % %%&LMM M"uww"D3F333r'c"|jjdkSNrFrrDr,s r%rzUntypedStorage.is_cudas{6))r'c"|jjdkSNrNr^r,s r%rzUntypedStorage.is_hpus{5((r'r(rBc*|S)zReturns the file name associated with this storage. The file name will be a string if the storage is on CPU and was created via :meth:`~torch.from_file()` with ``shared`` as ``True``. This attribute is ``None`` otherwise. )rr,s r%rzUntypedStorage.filenames!!###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`` )rr,r[s r%r,zUntypedStorage.share_memory_s"B%uww$d5f555r'c6tj|i|Sr)rr[r[s r%r[zUntypedStorage._share_fd_cpu_ s%uww%t6v666r'c6tj|i|Sr)rrYr[s r%rYz#UntypedStorage._share_filename_cpu_s+uww+T) r@rArBr2rErrrrWr,r[rYrFrGs@r%rrs44444 **X*))X)$$$X$" 6 6 6 6"! 6D"7777"!7"===="!=====r'cRtjtj|dS)NF) weights_only)rIloadrr)rs r%rrs :bjmm% 8 8 88r'cPtjtjtjtjtjtjtjtjtj tj tj tj tj tjtjhSr)rIrrrrfloat8_e8m0fnufloat4_e2m1fn_x2bits8bits16bits1x8bits2x4bits4x2 complex32uint16uint32uint64r r'r% _new_dtypesrts_                r'citjdtjdtjdtjdtjdtjdtjdtjdtj d tj d tj d tj d tj d tjdtjdtjdtjdS)N DoubleStorage FloatStorage HalfStorage LongStorage IntStorage ShortStorage CharStorage ByteStorage BoolStorageBFloat16StorageComplexDoubleStorageComplexFloatStorage QInt8Storage QInt32Storage QUInt8StorageQUInt4x2StorageQUInt2x4Storage)rIrrrrrint16rrrrr rqint8qint32quint8quint4x2quint2x4r r'r%_dtype_to_storage_type_mapr1s  o  ^   M   M    <    ^    M   ]   M  )   -   +   ^   o   o  )! " )# r'cZdtD}|S)Nci|]\}}|| Sr r )rrRvals r% z._storage_type_to_dtype_map..OsOOOhc3cOOOr')ritems) dtype_maps r%_storage_type_to_dtype_maprMs.OO*D*F*F*L*L*N*NOOOI r'c |tjtjtjtjtjfvrtjtjtjtjtjtjtjtjtjtji}tj ||||}ntj |||}| j S)Nr) rIrrrrrrint32rrr_untyped_storage)sequencerArinterpret_dtypes tmp_tensors r%_get_storage_from_sequencerSs       L%+ NEK NEK L%+ K  \ ,U3F   \(%GGG  $ $ & & 77r'c|tr!t|ttjfSt|tSr) HAS_NUMPYrrnpinteger)xs r%_isintrls1"!c2:.///!S!!!r'ctSr)"_always_warn_typed_storage_removalr r'r%&_get_always_warn_typed_storage_removalrvs --r'c|t|ts$tdt|j|adS)Nzalways_warn must be bool, got )rrrNrDr@r) always_warns r%&_set_always_warn_typed_storage_removalrzsJ k4 ( (  IT+->->-G I I   *5&&&r'r3cd}ts |r2d}tj|t|dzdtjd<dSdS)NcVttdsdStjd S)N has_warnedT)hasattr_warn_typed_storage_removal__dict__r r'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()r4) stacklevelTr)rwarningswarn UserWarningrr)rrmessages r%rrs{JJJ .//B==??B Q   g{zA~FFFF=A#,\:::BBr'c$dtjd<dS)NFr)rrr r'r%!_reset_warn_typed_storage_removalrs9>(666r'modulerc|ddd}|dtjdfvr|SdS)N.r4rFrNr)rsplitrI_C_get_privateuse1_backend_name)r last_parts r%_get_device_from_modulersE c1%%b)IVUXCCEEuMMMur'c|eZdZUdZded<dZded<ded<ed Zedsd Zd Z dddddfd Z ddddddZ edZ edZ dZdtdZdZdudZdZdZdZdZdvdwd!Zd"Zd#Z dxdyd&Zdxdtd'Zdxdtd(Zdd)dzd,Zd-Zd.Zd{d0Zd1Z d2Z!d3Z"d4Z#d5Z$d6Z%fd7Z&d8Z'd9Z(d:Z)dvd|d<Z*dvd|d=Z+d>Z,d?Z-dd@dAZ.edBZ/edCZ0dDZ1dEZ2dFZ3dGZ4dHZ5dIZ6dJZ7dKZ8dLZ9dMZ:e;dNZe;dddQdRZ?dSZ@dTZAdUZBdVZCdWZDdXZEdYZFdZZGd[ZHd\ZId]ZJd^ZKd_ZLd`ZMdaZNdbZOdcZPe;ddZQe;deZRdfZSdgZTdhZUdiZVdjZWdkZXe;dlZYdmZZdnZ[e;dd@doZ\dpZ]dqZ^drZ_xZ`S)}rFr rNrr torch.dtyperAc|jSr)rAr,s r%_dtypezTypedStorage._dtypes zr'r(rBc|jjS)zReturns 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.)rrr,s r%rzTypedStorage.filenames$--r'ct|td|||S)Nr)r_setitemslice_size)r"values r%fill_zTypedStorage.fill_s9#%%% eAtzz||,,e444 r') wrap_storagerAr _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 specifiedr4 Too many positional argumentsr Argument type not recognized: T)rArrB 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 rrAr)rrIr_LegacyStorage RuntimeErrorrr__new__rrr collectionsabcSequencerrDrrrArrrA) r^rrArrr# arg_error_msg cls_devicers r%rzTypedStorage.__new__s * ' ) ) ) %-. . .J  ,  77??3'' '666 !"!$UU "!$TT#t99q==&%(II IINN"47OO#&tAw 0HII#$%LT$q']]LLM $*23>BB" t99>>&%)) ",0DEE#%kW[\hWiWikkl 5S^DD &+z99&%@ @@%1%8%=@@@ $!-)" r')rrArrc,|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 givenrrrF/Cannot create CUDA storage with quantized dtyperjr4rr)rrrrrIrArrDrrget_default_dtyperrrrrrrrr6rrrr)r"rrArrr#rs r%r&zTypedStorage.__init__s * ' ) ) ) E   #4yyA~~"!%% }"!$JJeU[11 !RT%[[RRS !"!`a DJlE,@AA !gSWXdSeSeggh %1D ! ! !7%%vFFFz   ;&((&I4yyA~~(-(G;r5rr6 bool | Nonectt|tr!|j|j|n|j|||Sr)rrrrr:r9s r%r:zTypedStorage.copy_s]#%%% fl + + >  ! ' '(? N N N N  ! ' ' = = = r'cFt|Sr)r_nbytesr,s r%r>zTypedStorage.nbytes s#%%%||~~r'c4|jSr)rr>r,s r%rzTypedStorage._nbytess$++---r'r!_StorageBase | TypedStorage | strct|V|}||jdz|jzSd|jt |jgS|j||S)Nr)r_get_legacy_storage_classrAr@rrDr)r"rAr6 legacy_classs r%rDzTypedStorage.types $%%% =99;;L'#.4|7LLL88T_d4jj.ABCC C(--e\BB Br'ct|jtjtjtjtjtjfvrtd|j ||}| |S)Nr) rrArIrrrrrrrrFr)r"rr6 cuda_storages r%rFzTypedStorage.cuda%sx#%%% : L N N L K    PQQ Q,11&,GG ((666r'ct|jtjtjtjtjtjfvrtd|j ||}| |S)Nz.Cannot create HPU storage with quantized dtype) rrArIrrrrrrrrNr)r"rr6 hpu_storages r%rNzTypedStorage.hpu2sx#%%% : L N N L K    OPP P+// EE ((555r'rrrctt|tjstj|}|jtjtjtjtjtj fvr*td|j d|j ||}||S)NzCannot create z storage with quantized dtyperG)rrrIrrArrrrrrrDupperrrJr)r"rr6 to_storages r%rJzTypedStorage.to?s#%%%&%,// *\&))F : L N N L K    S!2!2!4!4SSS *--V,-WW ((444r'cFt|Sr)rr6r,s r%rPzTypedStorage.element_sizePs #%%%!!###r'cJtj|jSr)rIr5r6rAr,s r%r6zTypedStorage._element_sizeUs|))$*555r'r cPt|jSr)rrrSr,s r%rSzTypedStorage.get_deviceXs##%%%$//111r'c Ptdtjdjdjdt d }jjdkrd|zSdd fd t Dz}|d z|zS) Nrz(dtype=z , device=rrrrrrc3BK|]}t|VdSrrrs r%rz'TypedStorage.__str__..es-'Q'QDG 'Q'Q'Q'Q'Q'Qr'r) rrIrrArrrDrrr@rs` r%__str__zTypedStorage.__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__hs#%%%4yyr'cttfdtDS)Nc3(K|] }|V dSrr rs r%rz(TypedStorage.__iter__..nrr')rrrr@r,s`r%rzTypedStorage.__iter__lsB#%%%8888U499;;%7%7888888r'cvt|tj|jSr)rrcopyrr,s r%rzTypedStorage.__copy__ps/#%%%((43H)I)IJJJr'cHt||Sr)r _deepcopyr"rs r%rzTypedStorage.__deepcopy__ts #%%%~~d###r'c\|tj|j|Sr)rrdeepcopyrrs r%rzTypedStorage._deepcopyys%((t7Ld)S)STTTr'ctt|zSr)rrrr>rs r%rzTypedStorage.__sizeof__|s2#%%%ww!!##dkkmm33r'cvt||jS)r)rrrrr,s r%rzTypedStorage.clones1#%%%(()>)D)D)F)FGGGr'c<tt|Sr)rrr,s r%rzTypedStorage.tolists#%%%Dzzr'cvt||jS)r)rrrrr,s r%rzTypedStorage.cpus1#%%%(()>)B)B)D)DEEEr'rcRt|j|S)a0Determine whether the CPU TypedStorage is already pinned on device. Args: device (str or torch.device): The device to pin memory on (default: ``None``). This argument is discouraged and subject to deprecated. Returns: A boolean variable. )rrr r!s r%r zTypedStorage.is_pinneds' $%%%$..v666r'czt||j|S)a6Copy 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: ``None``). This argument is discouraged and subject to deprecated. Returns: A pinned CPU storage. rj)rrrr#r!s r%r#zTypedStorage.pin_memorys? $%%%((  ! , ,F , ; ;   r'cFt|S)r')r_share_memory_r,s r%r,zTypedStorage.share_memory_s #%%%""$$$r'c8|j|Sr)rr,r,s r%rzTypedStorage._share_memory_s ++--- r'rjc|d}tj|}tj||z|}t ||jdS)r.NrrjTr)rIrrr/r6rrA)r"r@rr$s r%r/zTypedStorage._new_sharedsp >Ff%%.:: 4%%'' ';  ( d    r'c|jjSr)rrr,s r%rzTypedStorage._cdatas$++r'c6t|jjSr)rrrr,s r%rzTypedStorage.devices#%%%$++r'cFt|Srrr,s r%r@zTypedStorage.sizerr'c^|j|zSr)rr>r6r,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)rrAKeyError)r"es r%r'z!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%rUzTypedStorage.data_ptrs#%%%~~r'c4|jSr)rrUr,s r%r.zTypedStorage._data_ptrs$--///r'cPt|jSr)rrrWr,s r%rWzTypedStorage.resizables##%%%$..000r'cLt||dSr)r_resize_rws r%rxzTypedStorage.resize_s&#%%% dr'cd|j||zdSr)rrxr6rws r%r2zTypedStorage._resize_s0 %%dT-?-?-A-A&ABBBBBr'c$tj|i|Sr)rrrds r%rzTypedStorage._free_weak_refs,d=f===r'c&|jj|i|Sr)rrzr!s r%rzzTypedStorage._weak_refs.t$.????r'c8t|j|i|Sr)r _from_bufferrds r%rezTypedStorage.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 rAzefrom_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) rrIrrrDrrrerr)r^rArr#r$r$s r%r7zTypedStorage._from_buffers* ,  16E+---EE\6>%%vFFF{e##"W&+WW5:4H4T5"5&,55OO  CIINN"N!"N JE#2>"&,OQUVVVVr'ct|tjstdt |tjg|j|j|| }| | kr| }|Sr) rrIrArrDrrrrJrrUrrs r%r zTypedStorage._to$s%-- XVeVVWW W L4:dk B B B T$ZZ RYY ^         0 0mmooGr'c\t|tjSr)rr rIrr,s r%rzTypedStorage.double1"#%%%xx %%%r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.float6"#%%%xx $$$r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.half;"#%%%xx ###r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.long@r@r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.intEs"#%%%xx """r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.shortJr>r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.charOr@r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.byteTr>r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.boolYr@r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.bfloat16^s"#%%%xx'''r'c\t|tjSr )rr rIr r,s r%r zTypedStorage.complex_doublecs"#%%%xx &&&r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.complex_floathr<r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.float8_e5m2ms##%%%xx)***r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.float8_e4m3fnrs##%%%xx+,,,r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.float8_e5m2fnuzw##%%%xx-...r'c\t|tjSr)rr rIrr,s r%rzTypedStorage.float8_e4m3fnuz|rMr'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) rrrrrrIr5r6rA)r^rrr@r$rs r%rzTypedStorage.from_filesv( $%%% ,  PQQ Q(22 fdU\%?%? %J%JJ  #?333r'c$tj|i|Sr)rrrds r%rzTypedStorage._expireds&7777r'c&|jj|i|Sr)rrur!s r%ruzTypedStorage._write_files0t$0$A&AAAr'c&|jj|i|Sr)rr|r!s r%r|zTypedStorage._set_from_file3t$3TDVDDDr'c&|jj|i|Sr)rr~r!s r%r~zTypedStorage._set_cdatas/t$/@@@@r'c&|jj|i|Sr)rrr!s r%rzTypedStorage._share_cuda_s1t$14B6BBBr'cFt|Sr)r _is_sharedr,s r%rzTypedStorage.is_shareds#%%%   r'c4|jSr)rrr,s r%rWzTypedStorage._is_shareds$..000r'c.tjj|i|Sr)rIrrrds r%rzTypedStorage._new_shared_cudas#4dEfEEEr'cb|jj|i|\}}}||||zfSr)rrYr6)r"r#r$manager_handlestorage_handler@s r%rYz!TypedStorage._share_filename_cpu_sH 7D ! 6 G G G    ~tt7I7I7K7K/KKKr'c8|j|Sr)rrsr,s r%rszTypedStorage._shared_decrefs ,,... r'c.tjj|i|SrrIrrlrms r%rnz!TypedStorage._release_ipc_counter#=tNvNNNr'c&|jj|i|Sr)rrr!s r%rzTypedStorage._shared_increfrSr'c^|jj|i|\}}|||zfSr)rr[r6)r"r#r$fdr@s r%r[zTypedStorage._share_fd_cpu_s<74(7HHHD44--/////r'cr|jtvrdSt|j}|jjdddtjfvrdS|jjdkrtntt|jj} t||S#t$rYdSwxYw)NrrFrN) rArrrDrIrrgetattrAttributeError)r" storage_namers r%rz&TypedStorage._get_legacy_storage_classs :799 9 94133DJ? ;     H 2 2 4 4 $   4[%..EEGE4;CS4T4T  6<00 0   44 sB(( B65B6r>r=)Fr)r5rr6rr<)rArBr6rr(r)rrr6rr(rr;r?)ar@rArBrrCrrErrrrr&rrr1rr-rr4rr2rr:r>rrDrFrNrJrPr6rSr rrrrrrrrrr r#r,rr/rrr@rr(r'rrUr.rWrxr2rDrrzrer7r rrrrrrrrrrr rrrrrrrrur|r~rrrWrrYrsrnrr[rrFrGs@r%rrs?I(,L,,,, X...X.  ZZZZZZZ~ YVYVYVYVYVv;;X;::X:%%% <<<< ****4)))! ! ! F"""*2*2*2X ... !"CCCCC" 7 7 7 7 7 6 6 6 6 6BG555555"$$$ 6662222 . . .999KKK$$$ UUU44444HHH  FFF 7 7 7 7 7      %%% +/      ,,X,,,X, FFF +++ KKK 333    000111 CCC>>[>@@@11[1'+DWWWW[W>   &&& %%% $$$ $$$ ### %%% $$$ %%% $$$ ((( ''' &&& +++ --- /// /// [888[8BBBEEEAAACCC!!! 111FF[FLLL04OOOO[OEEE000r'c eZdZUded<dZdS)_LegacyStorageMetarrAct|tur4t|j}||jjko|j|jkSdSr<)rDrrrArrA)r^instancers r%__instancecheck__z$_LegacyStorageMeta.__instancecheck__sL >>\ ) )0@@J(/"66 X^+ ur'N)r@rArBrCrlr r'r%riris1r'ricPeZdZedZedZedZdS)rctj||z}||S)r.r)rIrr/r6)r^r@r$s r%r/z_LegacyStorage._new_sharedsD .::4##%%BUBUBWBW;WXXs0000r'c.tjj|i|Srr_rds r%rnz#_LegacyStorage._release_ipc_counterr`r'c|tj|jz}|tj|||S)Nr)rIr5r6rArri)r^rgrhr@ bytes_sizes r%_new_shared_filenamez#_LegacyStorage._new_shared_filenamesTEL66syAAA s-FFj    r'N)r@rArBrDr/rnrrr r'r%rrsf11[1 OO[O  [   r'r) metaclassr(cx t|S#t$r}td|d|d}~wwxYw)Nzpickle storage type "z" is not recognized)rr*)r(r+s r%#_get_dtype_from_pickle_storage_typerus\)++,?@@  L$7 L L L   s 949)r3)rr)r(r)> __future__rrrrUrrKrtypingrrrrtyping_extensionsrrI torch._utilsr r torch.typesr r r torch._prims_commonr__all__numpyrrModuleNotFoundErrorLockrJrrCrrrWr StorageBaserrcachertrrrrrrrrrrr__doc__rDrFrNrJrirrur r'r%rs#""""""  444444444444"""""" ########,,,,,,,,,,3222222 + ,III BBB $Y^%%022222 GC4555@"@"@"@"@"@"@"@"F $$$N@=@=@=@=@=UX)<@=@=@=F999 , 6  8882"""&+"...555BBBB(???C C C C C C C C L"M (-5 '+3 +              \-?    *sA A('A(