a d@sddlZddlZddlmZmZddlmZddlmZm Z m Z m Z m Z ddl Z ddlZddlmZddlZzddlZdZWneydZYn0e dd d ZGd d d ZGd ddejjeZddZee_ee_eddddZeddddZddZ ddZ!da"ddZ#ddZ$d.d d!Z%d"d#Z&Gd$d%d%Z'ej(e'j_(ej(e'j_(Gd&d'd'eZ)Gd(d)d)e'e)d*Z*e+d+d,d-Z,dS)/N)_type_cuda)Storage)AnyTypeVarTypeUnioncast) lru_cacheTTz!Union[_StorageBase, TypedStorage])boundcseZdZUeed<dZeed<dZeed<ej ed<ddZ e dd d Z d d Z deeedddZe dddZe dddZdeeedddZdedddZe dddZe dddZe dddZd d!Zd"d#Zeeee ed$d%d&Zeeee ed$d'd(Zeedd)d*Zed d d+edd,d-Zeedd.d/Z eedd0d1Z!edd2d3Z"d4d5Z#e d6d7d8Z$edd9d:Z%edd;d<Z&d=d>Z'd?d@Z(dAdBZ)eddCdDZ*eeddEdFZ+dGdHZ,edIdJZ-e.dKdLZ/eeddMdNZ0eeddOdPZ1dQdRZ2dSdTZ3dUdVZ4dWdXZ5dYdZZ6d[d\Z7fd]d^Z8d_d`Z9dadbZ:dcddZ;dedfZdkdlZ?dmdnZ@dodpZAdqdrZ dsdtZBdudvZCdwdxZDdydzZd{d|ZEd}d~ZFddZGddZHddZIeddddZJddZKZLS) _StorageBase_cdataF is_sparse is_sparse_csrdevicecOsdSNselfargskwargsrrV/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/storage.py__init__z_StorageBase.__init__returncCsdSrrrrrr__len__rz_StorageBase.__len__cCsdSrrridxrrr __getitem__rz_StorageBase.__getitem__N)source non_blockingrcCsdSrrrr#r$rrrcopy_rz_StorageBase.copy_cCsdSrrrrrrnbytesrz_StorageBase.nbytescCs|Sr)r'rrrrsizesz_StorageBase.sizedtyper$rcCsdSrr)rr*r$rrrtype!rz_StorageBase.typecKsdSrr)rrr$rrrrcuda"rz_StorageBase.cudacCsdSrrrrrr element_size#rz_StorageBase.element_sizecCsdSrrrrrr get_device$rz_StorageBase.get_devicecCsdSrrrrrrdata_ptr%rz_StorageBase.data_ptrcOsdSrrrrrr_share_filename_cpu_(rz!_StorageBase._share_filename_cpu_cOsdSrrrrrr_share_fd_cpu_)rz_StorageBase._share_fd_cpu_)clsr(rcCsdSrrr2r(rrr_new_using_filename_cpu*sz$_StorageBase._new_using_filename_cpucCsdSrrr3rrr_new_using_fd_cpu,sz_StorageBase._new_using_fd_cpucOsdSrrr2rrrrr from_buffer.sz_StorageBase.from_buffer)rr*cCsdSrr)r2managerobjr(rr*rrr_new_shared_filename_cpu0sz%_StorageBase._new_shared_filename_cpucOsdSrrr6rrr_release_ipc_counter_cuda2sz&_StorageBase._release_ipc_counter_cudacOsdSrrr6rrr_new_with_weak_ptr4sz_StorageBase._new_with_weak_ptrcCsdSrrrrrr_shared_decref6rz_StorageBase._shared_decrefcOsdSrrrrrr _write_file7rz_StorageBase._write_file)r(cCsdSrrrr(rrrresize_8rz_StorageBase.resize_cOsdSrrrrrr _weak_ref9rz_StorageBase._weak_refcCsdSrrrrrr is_pinned:rz_StorageBase.is_pinnedcOsdSrrrrrr_set_from_file;rz_StorageBase._set_from_filecOsdSrrrrrr _set_cdata<rz_StorageBase._set_cdatacOsdSrrrrrr _share_cuda_=rz_StorageBase._share_cuda_cCsdSrrrrrr is_shared>rz_StorageBase.is_sharedcOsdSrrr6rrr_new_shared_cuda?sz_StorageBase._new_shared_cudacOsdSrrrrrr_shared_increfArz_StorageBase._shared_increfcOsdSrrr6rrr_free_weak_refBsz_StorageBase._free_weak_refcCsdSrrrrrris_cudaDsz_StorageBase.is_cudacCsdSrr)r2filenamesharedr'rrr from_fileFsz_StorageBase.from_filecOsdSrrr6rrr_expiredHsz_StorageBase._expiredcsldtdjdtd}jjdkr8d|Sddfd d tD}|d |SdS) N[z(device= ) of size ]meta...   c3s|]}t|VqdSrstr.0irrr Rrz'_StorageBase.__str__.. )torchtypenamerlenr+joinranger(rZinfo_strZdata_strrrr__str__Ks $z_StorageBase.__str__cCst|SrrVrrrr__repr__Usz_StorageBase.__repr__csttfddtS)Ncs|SrrrZrrrYrz'_StorageBase.__iter__..)itermaprar(rrrr__iter__Xsz_StorageBase.__iter__cCs|Sr)clonerrrr__copy__[sz_StorageBase.__copy__cCs6|di}|j|vr ||jS|}|||j<|S)Nr]) setdefaultrrj)rmemoZ new_storagerrr __deepcopy__^s     z_StorageBase.__deepcopy__cCs&t}tj||ddt|ffSNF)Z_use_new_zipfile_serializationioBytesIOr]save_load_from_bytesgetvaluerbrrr __reduce__fsz_StorageBase.__reduce__cst|Sr)super __sizeof__r(r __class__rrrzksz_StorageBase.__sizeof__cCst|||jd|S)Returns a copy of this storager)r+r'rr&rrrrrjnsz_StorageBase.clonecCst|Sz6Returns a list containing the elements of this storage)listrrrrtolistrsz_StorageBase.tolistcCs*|jjdkr"t||dS|SdS)AReturns a CPU copy of this storage if it's not already on the CPUcpuFNrr+r]UntypedStorager(r&rrrrrvs z_StorageBase.cpucCs.|jjdkr&tj|dd|dS|SdS)rmpsr~FNrrrrrr}s z_StorageBase.mpscCsdt|tjstdt|tjgtj|jdt t | | }| | kr`|}|SNz*Argument 'dtype' must be torch.dtype, not r*r) isinstancer]r* TypeErrorr+tensoruint8rset_r rto_typed_storager/rjrr*storagerrr_tos  *z_StorageBase._tocCs |tjSz!Casts this storage to double type)rr]doublerrrrrsz_StorageBase.doublecCs |tjSz Casts this storage to float type)rr]floatrrrrrsz_StorageBase.floatcCs |tjSzCasts this storage to half type)rr]halfrrrrrsz_StorageBase.halfcCs |tjSzCasts this storage to long type)rr]longrrrrrsz_StorageBase.longcCs |tjSzCasts this storage to int type)rr]intrrrrrsz_StorageBase.intcCs |tjSz Casts this storage to short type)rr]shortrrrrrsz_StorageBase.shortcCs |tjSzCasts this storage to char type)rr]int8rrrrcharsz_StorageBase.charcCs |tjSzCasts this storage to byte type)rr]rrrrrbytesz_StorageBase.bytecCs |tjSzCasts this storage to bool type)rr]boolrrrrrsz_StorageBase.boolcCs |tjSz#Casts this storage to bfloat16 type)rr]bfloat16rrrrrsz_StorageBase.bfloat16cCs |tjSz)Casts this storage to complex double type)rr]cdoublerrrrcomplex_doublesz_StorageBase.complex_doublecCs |tjSz(Casts this storage to complex float type)rr]cfloatrrrr complex_floatsz_StorageBase.complex_floatcCsH|jrtd|dddl}|jj}t|||d|S)z@Copies the storage to pinned memory, if it's not already pinned.z cannot pin 'z' only CPU memory can be pinnedrN) allocator) rJrr+Z torch.cudar,ZmemoryZ_host_allocatorr(r&)rr]rrrr pin_memorys  z_StorageBase.pin_memorycCs4ddlm}|jrn|dkr(|n||S)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. Returns: self rget_sharing_strategy file_system)torch.multiprocessingrrJr0r1)rrrrr share_memory_s   z_StorageBase.share_memory_rr~cCsNddlm}t|}|jdkr,|||dS|dkr@||S||SdS)>Creates a new storage in shared memory with the same data typerrr,r~rN)rrr]rr+r4r5)r2r(rrrrr _new_shareds      z_StorageBase._new_sharedcCs|Srrrrrruntypedsz_StorageBase.untyped)N)NF)NF)M__name__ __module__ __qualname__r__annotations__rrrr]rrrrr"r r&r'r(rWr+r,r-r.r/r0r1 classmethodrr4r5r7r:r;r<r=r>r@rArBrCrDrErFrGrHrIpropertyrJrMrNrcrdrirkrnrxrzrjrrrrrrrrrrrrrrrrrr __classcell__rrr{rrs         rcs(eZdZfddZeddZZS)rcs&|jjdkrtdtj|i|S)NrR$Not available for 'meta' device type)rr+NotImplementedErrorryr"rr{rrr"s zUntypedStorage.__getitem__cCs |jjdkSNr,)rr+rrrrrJszUntypedStorage.is_cuda)rrrr"rrJrrrr{rrs rcCstt|Sr)r]loadrqrr)rwrrrrtsrt)maxsizec"Csjtjdtjdtjdtjdtjdtjdtjdtjdtj d tj d tj d tj d tj d tjdtjdtjdtjdiS)NZ DoubleStorageZ FloatStorageZ HalfStorageZ LongStorageZ IntStorageZ ShortStorageZ CharStorageZ ByteStorageZ BoolStorageZBFloat16StorageZComplexDoubleStorageZComplexFloatStorageZ QInt8StorageZ QInt32StorageZ QUInt8StorageZQUInt4x2StorageZQUInt2x4Storage)r]rrrrrint16rrrrrrqint8qint32quint8quint4x2quint2x4rrrr_dtype_to_storage_type_maps$rcCsddtD}|S)NcSsi|]\}}||qSrr)rYkeyvalrrr sz._storage_type_to_dtype_map..)ritems)Z dtype_maprrr_storage_type_to_dtype_mapsrc Csx|tjtjtjtjtjfvr^tjtjtjtjtjtjtjtjtjtji}tj ||||d}ntj |||d}| j S)Nr) r]rrrrrrint32rrr_untyped_storage)sequencer*rinterpret_dtypes tmp_tensorrrr_get_storage_from_sequences$rcCs"trt|ttjfSt|tSdSr) HAS_NUMPYrrnpinteger)xrrr_isint/srFcCstSr)"_always_warn_typed_storage_removalrrrr&_get_always_warn_typed_storage_removal7srcCst|tsJ|adSr)rrr)Z always_warnrrr&_set_always_warn_typed_storage_removal:srcCs:dd}ts|r6d}tj|t|dddtjd<dS)NcSsttdsdStjd SdS)N has_warnedT)hasattr_warn_typed_storage_removal__dict__rrrr is_first_timeBs z2_warn_typed_storage_removal..is_first_timeaTypedStorage 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)rrmessagerrrr?s  rcCsdtjd<dS)NFr)rrrrrr!_reset_warn_typed_storage_removalRsrcs&eZdZUdZejed<eddZddZ dddddfd d Z ddddd d d Z eddZ ddZ ddZddZdddZddZddZddZddZdeed d!d"Zd#d$Zd%d&Zdeeeeefd'd(d)Zded*d+d,Zd-d.Zd/d0Ze d*d1d2Z!d3d4Z"d5d6Z#d7d8Z$d9d:Z%d;d<Z&d=d>Z'fd?d@Z(dAdBZ)dCdDZ*dEdFZ+dGdHZ,dIdJZ-dKdLZ.ddMdNdOZ/edPdQZ0edRdSZ1dTdUZ2dVdWZ3dXdYZ4dZd[Z5d\d]Z6d^d_Z7d`daZ8dbdcZ9dddeZ:e;dfdgZe;dddldmdnZ?dodpZ@dqdrZAdsdtZBdudvZCdwdxZDdydzZ d{d|ZEd}d~ZFddZGddZddZHddZIddZJe;ddZKe;ddZLddZMddZNddZOddZPddZQddZRddZSe;ddZTddZUddZVe;ddMddZWddZXddZYddZZZ[S) TypedStorageFr*cCs|jSr)r*rrrr_dtypeZszTypedStorage._dtypecCs t|td|||S)Nr)r_setitemslice_size)rvaluerrrfill_^szTypedStorage.fill_N) wrap_storager*r _internalcs~|s t|tjjkrtd|tkr2t|S|d}|durPt|d|durdt|d|durt|dkrt|dt|dkrt |dst |dt j j st|dt|dt||j|jd krd nd d d St|dkrt|dt |tjs(t|dt||jd kr8d nd }|jj|krft|d|d|jjt|||jd dSdS)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: torch.cudar,rT)r*rrB 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 rr*r)rr]r_LegacyStorage RuntimeErrorrry__new__r_rr collectionsabcSequencerr+rrrrr*)r2rr*rrr arg_error_msg cls_devicer{rrrcs   * zTypedStorage.__new__)rr*rrcGs|s td}|durt|dkr.t|d|durBt|dt|tjsdt|dt||durxt|d||_t|tjst|dt|||_ n|durt n||_t |durdn|}|jtj tj tjtjtjfvr |jd kr td t|dkr(tj|d |_ nt|d krt|drftjt|d||d |_ nDt|dtjjrt|d|j||_ nt|d t|dn t|ddS)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 dtyper~rrr)rr_rrr]r*rr+rrget_default_dtyperrrrrrrr _element_sizerrrr)rrr*rrrrrrrrsv        "zTypedStorage.__init__cCst|jjjdkSr)rrrr+rrrrrJszTypedStorage.is_cudacCs t|jS)z2Returns the internal :class:`torch.UntypedStorage`)rrrrrrrszTypedStorage.untypedcCs@t|tjksJt|tkr.t||jddSt||dSdS)NTrr)r+r]rrr*)runtyped_storagerrr_new_wrapped_storages z!TypedStorage._new_wrapped_storagecCst|SrrrrrrrrszTypedStorage.__len__cCs|dur|r|SdSnt|tkrBtdt|dt||r||ks`|| krxtd|d||dkr|S||Sn>||ks|| krtd|d|||SdS)Nrcan't index a  with zindex z" out of range for storage of size )rr+rr IndexErrorr()rr!Zis_stoprrr_maybe_wrap_indexs* zTypedStorage._maybe_wrap_indexcCst|||Sr)rr)rr!rrrr __setitem__!szTypedStorage.__setitem__c Cst|ttfs*tdt|dt|t|rFtdt||jtjtj tj tj tj fvrtjtj tj tj tj tj tj tjtj tji}||j}tjg||jjd}|t|j|ddntjg|j|jjd|}|||<dS)Nr r z cannot set item with value type rTr)rrrrr+r]Z is_storager*rrrrrrrrrrrrr)rr!rrZ tmp_dtyperrrrr%s(   zTypedStorage._setitemcCst||Sr)r_getitemr rrrr"=szTypedStorage.__getitem__c Cs|jjjdkrtdt|tr*tdn&t|tsPtdt|dt||jt j t j t j t j t jfvrt j t jt j t jt j t jt j t jt jt ji}t|j||jdd|S||}t jg|j|jjd|}||S) NrRrz7slices are only supported in UntypedStorage.__getitem__r r Trr)rrr+rrrrrr*r]rrrrrrrrrrr rritem)rr!rZ idx_wrappedrrrrrAs.    zTypedStorage._getitem)r#r$cCs4tt|tr"|j|j|n|j|||Sr)rrrrr&r%rrrr&_s  zTypedStorage.copy_cCst|Sr)r_nbytesrrrrr'gszTypedStorage.nbytescCs |jSr)rr'rrrrrlszTypedStorage._nbytesr)cCsVt|durD|}|dur.|jd|jSd|jt|jgS|j||SdS)N.)r_get_legacy_storage_classrrr`r+r)rr*r$Z legacy_classrrrr+oszTypedStorage.typercKsLt|jtjtjtjtjtjfvr,td|j j ||fi|}| |S)Nr) rr*r]rrrrrrrr,r)rrr$rZ cuda_storagerrrr,|s zTypedStorage.cudacCst|Sr)rrrrrrr-szTypedStorage.element_sizecCstj|jSr)r]_utilsrr*rrrrrszTypedStorage._element_sizecCst|jSr)rrr.rrrrr.szTypedStorage.get_devicec sztdtdjdjdtd }jjdkrFd|Sdd fd d t D}|d |SdS) NrOz(dtype=z , device=rPrQrRrSrTrUc3s|]}t|VqdSrrVrXrrrr[rz'TypedStorage.__str__..r\) rr]r^r*rr_r+r`rar(rbrrrrcs $zTypedStorage.__str__cCstt|Sr)rrWrrrrrdszTypedStorage.__repr__cs$tttfddtS)Ncs|Srrrerrrrfrz'TypedStorage.__iter__..)rrgrhrar(rrrrriszTypedStorage.__iter__cCst|t|jSr)rrcopyrrrrrrkszTypedStorage.__copy__cCst||Sr)r _deepcopyrrmrrrrnszTypedStorage.__deepcopy__cCs|t|j|Sr)rrdeepcopyrrrrrrszTypedStorage._deepcopycstt|Sr)rryrzr'rr{rrrzszTypedStorage.__sizeof__cCst||jS)r})rrrrjrrrrrjszTypedStorage.clonecCstt|Sr)rrrrrrrszTypedStorage.tolistcCst||jS)r)rrrrrrrrrszTypedStorage.cpucCst||jS)zBCoppies the storage to pinned memory, if it's not already pinned.)rrrrrrrrrszTypedStorage.pin_memorycCst|S)r)r_share_memory_rrrrrs zTypedStorage.share_memory_cCs|j|Sr)rrrrrrrs zTypedStorage._share_memory_r~cCs>|dur d}t|}tjj|||d}t||jddS)rNrr~Tr)r]rrrrrr*)rr(rrrrrrs zTypedStorage._new_sharedcCs|jjSr)rrrrrrrszTypedStorage._cdatacCst|jjSr)rrrrrrrrszTypedStorage.devicecCst|Srrrrrrr(szTypedStorage.sizecCs|j|Sr)rr'rrrrrrszTypedStorage._sizecCst|Sr)r_pickle_storage_typerrrrpickle_storage_typesz TypedStorage.pickle_storage_typec CsLzt|jWStyF}z td|jd|WYd}~n d}~00dS)Nzdtype z is not recognized)rr*KeyError)rerrrrsz!TypedStorage._pickle_storage_typecCs&t}tj||ddt|ffSrorprvrrrrxszTypedStorage.__reduce__cCst|Sr)r _data_ptrrrrrr/szTypedStorage.data_ptrcCs |jSr)rr/rrrrr szTypedStorage._data_ptrcCst||dSr)r_resize_r?rrrr@ szTypedStorage.resize_cCs|j||dSr)rr@rr?rrrrszTypedStorage._resize_cOstj|i|Sr)rrIr6rrrrIszTypedStorage._free_weak_refcOs|jj|i|Sr)rrArrrrrAszTypedStorage._weak_refcOst|j|i|Sr)r _from_bufferr6rrrr7szTypedStorage.from_bufferrcOs|tkrd|durtn|}t|dur,dn|}|jdkrLtd|jtjj|d|i|}nH|dusxt|dkrtd|durtd|j }tjj|d|i|}t||ddS) Nrz3TypedStorage.from_buffer: Not available for device r*zefrom_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) rr]rrr+rrr7r_r)r2r*rrrrrrrr!s* zTypedStorage._from_buffercCs^t|tjstdt|tjg|j|jd|| }| | krZ| }|Sr) rr]r*rr+rrrrrr/rjrrrrr<s  $zTypedStorage._tocCst|tjSr)rrr]rrrrrrDszTypedStorage.doublecCst|tjSr)rrr]rrrrrrIszTypedStorage.floatcCst|tjSr)rrr]rrrrrrNszTypedStorage.halfcCst|tjSr)rrr]rrrrrrSszTypedStorage.longcCst|tjSr)rrr]rrrrrrXszTypedStorage.intcCst|tjSr)rrr]rrrrrr]szTypedStorage.shortcCst|tjSr)rrr]rrrrrrbszTypedStorage.charcCst|tjSr)rrr]rrrrrrgszTypedStorage.bytecCst|tjSr)rrr]rrrrrrlszTypedStorage.boolcCst|tjSr)rrr]rrrrrrqszTypedStorage.bfloat16cCst|tjSr)rrr]rrrrrrvszTypedStorage.complex_doublecCst|tjSr)rrr]rrrrrr{szTypedStorage.complex_floatcCs@t|tkrtdt|||tj|j}||d}|S)a from_file(filename, shared=False, size=0) -> Storage 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 size (int): number of elements in the storage z/from_file can only be called on derived classesr) rrrrrMr]rrr*)r2rKrLr(rrrrrrMs zTypedStorage.from_filecOstj|i|Sr)rrNr6rrrrNszTypedStorage._expiredcCst|jSr)rrrBrrrrrBszTypedStorage.is_pinnedcOs|jj|i|Sr)rr>rrrrr>szTypedStorage._write_filecOs|jj|i|Sr)rrCrrrrrCszTypedStorage._set_from_filecOs|jj|i|Sr)rrDrrrrrDszTypedStorage._set_cdatacOs|jj|i|Sr)rrErrrrrEszTypedStorage._share_cuda_cCst|Sr)r _is_sharedrrrrrFszTypedStorage.is_sharedcCs |jSr)rrFrrrrr!szTypedStorage._is_sharedcOstjj|i|Sr)r]rrGr6rrrrGszTypedStorage._new_shared_cudacOs*|jj|i|\}}}||||fSr)rr0r)rrrZmanager_handleZstorage_handler(rrrr0sz!TypedStorage._share_filename_cpu_cCs|j|Sr)rr=rrrrr=s zTypedStorage._shared_decrefcOstjj|i|Srr]rr;)r2rrrrrr_release_ipc_countersz!TypedStorage._release_ipc_countercOs|jj|i|Sr)rrHrrrrrHszTypedStorage._shared_increfcOs&|jj|i|\}}|||fSr)rr1r)rrrfdr(rrrr1szTypedStorage._share_fd_cpu_cCsh|jtvrdSt|j}|jjdvr,dS|jjdkrrCrDrErFr!rGr0r=r#rHr1rrrrr{rrUs   HD               rc@s eZdZUejed<ddZdS)_LegacyStorageMetar*cCs:t|tkr6|jdkrdnd}||jjko4|j|jkSdS)Nrr,rF)r+rrrr*)r2instancerrrr__instancecheck__s z$_LegacyStorageMeta.__instancecheck__N)rrrr]r*rr*rrrrr(s  r(c@s0eZdZeddZeddZeddZdS)rcCs tj||}||dS)rr)r]rrr)r2r(rrrrrsz_LegacyStorage._new_sharedcOstjj|i|Srr"r6rrrr#sz#_LegacyStorage._release_ipc_countercCs(|tj|j}|tj|||dS)Nr)r]rrr*rr:)r2r8r9r(Z bytes_sizerrr_new_shared_filenamesz#_LegacyStorage._new_shared_filenameN)rrrrrr#r+rrrrrs   r) metaclass)rc CsHz t|WStyB}ztd|d|WYd}~n d}~00dS)Nzpickle storage type "z" is not recognized)rr)rrrrr#_get_dtype_from_pickle_storage_types  r-)r)-rqr]rrrZ torch.typesrtypingrrrr r rr functoolsr rnumpyrrModuleNotFoundErrorr rZ_CZ StorageBaserrtr+r,rrrrrrrrrr__doc__r(rrWr-rrrrsT     U