a 7Sicx@s^ddlZddlZddlmZmZddlmZddlmZm Z m Z m Z m Z ddl Z ddlZddlmZzddlZdZWneydZYn0e dd d ZGd d d eZGd ddejjeZddZee_ee_eddddZeddddZddZ ddZ!GdddZ"ej#e"j_#ej#e"j_#GdddeZ$Gddde"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/dMdNZ0dOdPZ1dQdRZ2dSdTZ3dUdVZ4dWdXZ5fdYdZZ6d[d\Z7d]d^Z8d_d`Z9dadbZ:dcddZ;dedfZdkdlZ dmdnZ?dodpZ@dqdrZAdsdtZdudvZBdwdxZCdydzZDd{d|ZEd}d~ZFeddddZGddZHZIS) _StorageBase_cdataF is_sparse is_sparse_csrdevicecOsdSNselfargskwargsrrI/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/storage.py__init__z_StorageBase.__init__returncCsdSrrrrrr__len__rz_StorageBase.__len__cCsdSrr)ridxrrr __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'rcCsdSrrr1r'rrr_new_using_filename_cpu)sz$_StorageBase._new_using_filename_cpucCsdSrrr2rrr_new_using_fd_cpu+sz_StorageBase._new_using_fd_cpucOsdSrrr1rrrrr from_buffer-sz_StorageBase.from_buffer)rr)cCsdSrr)r1managerobjr'rr)rrr_new_shared_filename_cpu/sz%_StorageBase._new_shared_filename_cpucOsdSrrr5rrr_release_ipc_counter_cuda1sz&_StorageBase._release_ipc_counter_cudacOsdSrrr5rrr_new_with_weak_ptr3sz_StorageBase._new_with_weak_ptrcCsdSrrrrrr_shared_decref5rz_StorageBase._shared_decrefcOsdSrrrrrr _write_file6rz_StorageBase._write_file)r'cCsdSrrrr'rrrresize_7rz_StorageBase.resize_cOsdSrrrrrr _weak_ref8rz_StorageBase._weak_refcCsdSrrrrrr is_pinned9rz_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_sharedcOsdSrrr5rrr_new_shared_cuda>sz_StorageBase._new_shared_cudacOsdSrrrrrr_shared_incref@rz_StorageBase._shared_increfcOsdSrrr5rrr_free_weak_refAsz_StorageBase._free_weak_refcCsdSrrrrrris_cudaCsz_StorageBase.is_cudacsLddfddtD}|dtdjdtdS) N  c3s|]}t|VqdSrstr.0irrr Grz'_StorageBase.__str__.. [z(device= ) of size ])joinranger'torchtypenamerlenrZdata_strrrr__str__Fs $z_StorageBase.__str__cCst|SrrLrrrr__repr__Lsz_StorageBase.__repr__csttfddtS)Ncs|SrrrPrrrPrz'_StorageBase.__iter__..itermaprVr'rrrr__iter__Osz_StorageBase.__iter__cCs|Sr)clonerrrr__copy__Rsz_StorageBase.__copy__cCs6|di}|j|vr ||jS|}|||j<|S)NrW) setdefaultrrc)rmemo new_storagerrr __deepcopy__Us     z_StorageBase.__deepcopy__cCs&t}tj||ddt|ffSNF)Z_use_new_zipfile_serializationioBytesIOrWsave_load_from_bytesgetvaluerbrrr __reduce__]sz_StorageBase.__reduce__cstt||Sr)superr __sizeof__r'r __class__rrrtbsz_StorageBase.__sizeof__cCst|||jd|S)Returns a copy of this storager)r*r&rr%rrrrrcesz_StorageBase.clonecCst|Sz6Returns a list containing the elements of this storagelistrrrrtolistisz_StorageBase.tolistcCs*|jjdkr"t||dS|SdS)AReturns a CPU copy of this storage if it's not already on the CPUcpuFN)rr*rW_UntypedStorager'r%rrrrr~ms z_StorageBase.cpucCsdt|tjstdt|tjgtj|jdt t | | }| | kr`|}|SNz*Argument 'dtype' must be torch.dtype, not r)r) isinstancerWr) TypeErrorr*tensoruint8rset_r rtostorager.rcrr)rrrr_tots  *z_StorageBase._tocCs |tjSz!Casts this storage to double typerrWdoublerrrrr|sz_StorageBase.doublecCs |tjSz Casts this storage to float typerrWfloatrrrrrsz_StorageBase.floatcCs |tjSzCasts this storage to half typerrWhalfrrrrrsz_StorageBase.halfcCs |tjSzCasts this storage to long typerrWlongrrrrrsz_StorageBase.longcCs |tjSzCasts this storage to int typerrWintrrrrrsz_StorageBase.intcCs |tjSz Casts this storage to short typerrWshortrrrrrsz_StorageBase.shortcCs |tjSzCasts this storage to char typerrWint8rrrrcharsz_StorageBase.charcCs |tjSzCasts this storage to byte typerrWrrrrrbytesz_StorageBase.bytecCs |tjSzCasts this storage to bool typerrWboolrrrrrsz_StorageBase.boolcCs |tjSz#Casts this storage to bfloat16 typerrWbfloat16rrrrrsz_StorageBase.bfloat16cCs |tjSz)Casts this storage to complex double typerrWcdoublerrrrcomplex_doublesz_StorageBase.complex_doublecCs |tjSz(Casts this storage to complex float typerrWcfloatrrrr 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) rIrr*Z torch.cudar+ZmemoryZ_host_allocatorr'r%)rrWrrrr 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.multiprocessingrrIr/r0)rrrrr share_memory_s   z_StorageBase.share_memory_r~rxcCsNddlm}t|}|jdkr,|||dS|dkr@||S||SdS)>Creates a new storage in shared memory with the same data typerrr+rxrN)rrrWrr*r3r4)r1r'rrrrr _new_shareds      z_StorageBase._new_sharedcCs|Srrrrrr_untypedsz_StorageBase._untyped)N)NF)NF)J__name__ __module__ __qualname__r__annotations__rrrrWrrrrr!r r%r&r'rMr*r+r,r-r.r/r0 classmethodrr3r4r6r9r:r;r<r=r?r@rArBrCrDrErFrGrHpropertyrIr[r\rbrdrhrrrtrcr|r~rrrrrrrrrrrrrrr __classcell__rrrurrs        rc@seZdZeddZdS)rcCs |jjdkSNr+rr*rrrrrIsz_UntypedStorage.is_cudaN)rrrrrIrrrrrsrcCstt|Sr)rWloadrkrl)rqrrrrnsrn)maxsizec"Csjtjdtjdtjdtjdtjdtjdtjdtjdtj d tj d tj d tj d tj d tjdtjdtjdtjdiS)N DoubleStorage FloatStorage HalfStorage LongStorage IntStorage ShortStorage CharStorage ByteStorage BoolStorageBFloat16StorageComplexDoubleStorageComplexFloatStorage QInt8Storage QInt32Storage QUInt8StorageQUInt4x2StorageQUInt2x4Storage)rWrrrrrint16rrrrrrqint8qint32quint8quint4x2quint2x4rrrr_dtype_to_storage_type_maps$rcCsddtD}|S)NcSsi|]\}}||qSrr)rOkeyvalrrr sz._storage_type_to_dtype_map..)ritems)Z dtype_maprrr_storage_type_to_dtype_mapsrc Csz|tjtjtjtjtjfvr^tjtjtjtjtjtjtjtjtjtji}tj ||||d}ntj |||d}| S)Nr) rWrrrrrrint32rrrr)sequencer)rinterpret_dtypes tmp_tensorrrr_get_storage_from_sequences$rcCs"trt|ttjfSt|tSdSr) HAS_NUMPYrrnpinteger)xrrr_isintsrcseZdZUdZejed<ddZddddfdd Zdddd d d Z e d d Z ddZ ddZ ddZdddZddZddZdeedddZddZdeeeeefdd d!Zded"d#d$Zd%d&Zed"d'd(Zd)d*Zd+d,Zd-d.Zd/d0Z d1d2Z!fd3d4Z"d5d6Z#d7d8Z$d9d:Z%d;d<Z&d=d>Z'dd?d@dAZ(e dBdCZ)e dDdEZ*dFdGZ+dHdIZ,dJdKZ-dLdMZ.dNdOZ/e0dPdQZ1dRdSZ2e0dddTdUdVZ3dWdXZ4dYdZZ5d[d\Z6d]d^Z7d_d`Z8dadbZdcddZ9dedfZ:dgdhZ;didjZdkdlZe0dqdrZ?e0dsdtZ@dudvZAdwdxZBdydzZCd{d|ZDd}d~ZEddZFe0ddZGddZHddZIe0dd?ddZJddZKddZLddZMZNS) _TypedStorageFr)cCs||dt|<|S)NrrY)rvaluerrrfill_'sz_TypedStorage.fill_N) wrap_storager)rcsv|tjjkrtd|tkr(t|S|d}|durFt|d|durZt|d|durt|dkrzt|dt|dkrt|dst |dt j j st |dt|dt||jt|jtjurd nd d St|dkrt|d t |tjs"t |d t||jdkr2d nd }|jj|kr`t|d|d|jjt|||jdSdS)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: r+r~rB No positional arguments should be given when using 'wrap_storage': Argument 'wrap_storage' must be _UntypedStorage, but got torch.cudaz" Device of 'wrap_storage' must be z , but got rr))rWr_LegacyStorage RuntimeErrorrrs__new__rYrr collectionsabcSequencerr*r)evalrr+rr)r1rr)rr arg_error_msg cls_devicerurrr+s|   * z_TypedStorage.__new__)rr)rcGsd}|durt|dkr$t|d|dur8t|dt|tjsZt|dt||durnt|d||_t|tjst|dt|||_n |durt n||_t |durdn|}|jtj tj tj tjtjfvr|jd krtd t|dkrtj|d |_nt|d krt|drXtjt|d||d |_nDt|dtjjrt|d|j||_nt|d t|dn t|ddS)Na3_TypedStorage.__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 givenrr~r+/Cannot create CUDA storage with quantized dtyperxrrr)rYrrrWr)rr*r_storageget_default_dtyperrrrrrrrr,rrrr)rrr)rrrrrrrnsr       "z_TypedStorage.__init__cCs |jjdkSrrrrrrrIsz_TypedStorage.is_cudacCs|jSr)rrrrrrsz_TypedStorage._untypedcCs>t|tjksJt|tkr,t||jdSt||dSdS)Nrr)r*rWrrr))runtyped_storagerrr_new_wrapped_storages z"_TypedStorage._new_wrapped_storagecCs|j|Sr)rr&r,rrrrrsz_TypedStorage.__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 )r'r*rr IndexError)rr Zis_stoprrr_maybe_wrap_indexs* z_TypedStorage._maybe_wrap_indexc Cst|ttfs*tdt|dt||jtjtjtj tj tj fvrtjtj tjtj tj tj tj tj tj tji}||j}tjg||jdt|j|d}ntjg|j|jd|}|||<dS)Nr r rr)rrslicerr*r)rWrrrrrrrrrrrrr)rr rrZ tmp_dtyperrrr __setitem__s   z_TypedStorage.__setitem__c Cst|trtdn&t|ts:tdt|dt||jtjtjtj tj tj fvrtjtj tjtj tj tj tj tj tj tji}t|j||jd|S||}tjg|j|jd|}||S)Nz8slices are only supported in _UntypedStorage.__getitem__r r rr)rr rrr*r)rWrrrrrrrrrrr rrritem)rr rZ idx_wrappedrrrrr!s(    z_TypedStorage.__getitem__)r"r#cCs|j|||Sr)rr%rr$rrrr% sz_TypedStorage.copy_cCs |jSr)rr&rrrrr&sz_TypedStorage.nbytesr(cCsP|dur>|}|dur(|jd|jSd|jt|jgS|j||SdS)N.)_get_legacy_storage_classrrrUr*r)rr)r#Z legacy_classrrrr*s z_TypedStorage.typercKsF|jtjtjtjtjtjfvr&td|jj ||fi|}| |S)Nr) r)rWrrrrrrrr+r)rrr#rZ cuda_storagerrrr+sz_TypedStorage.cudacCstj|jSr)rW_utils _element_sizer)rrrrr,#sz_TypedStorage.element_sizecCs |jSr)rr-rrrrr-&sz_TypedStorage.get_devicec sTddfddtD}|dtdjdjdtd S) NrJrKc3s|]}t|VqdSrrLrNrrrrQ*rz(_TypedStorage.__str__..rRz(dtype=z , device=rSrT)rUrVr'rWrXr)rrYrZrrrr[)s$z_TypedStorage.__str__cCst|SrrLrrrrr\/sz_TypedStorage.__repr__csttfddtS)Ncs|Srrr]rrrr^3rz(_TypedStorage.__iter__..r_rrrrrb2sz_TypedStorage.__iter__cCs|t|jSr)rcopyrrrrrrd5sz_TypedStorage.__copy__cCs|t|j|Sr)rrdeepcopyr)rrfrrrrh8sz_TypedStorage.__deepcopy__cstt||Sr)rsrrtr&rrurrrt;sz_TypedStorage.__sizeof__cCs||jS)rw)rrrcrrrrrc>sz_TypedStorage.clonecCst|Sryrzrrrrr|Bsz_TypedStorage.tolistcCs||jS)r})rrr~rrrrr~Fsz_TypedStorage.cpucCs||jS)zBCoppies the storage to pinned memory, if it's not already pinned.)rrrrrrrrJsz_TypedStorage.pin_memorycCs|j|S)r)rrrrrrrNs z_TypedStorage.share_memory_rxcCs<|dur d}t|}tjj|||d}t||jdS)rNr~rxr)rWrrrr,rr))rr'rrrrrrZs z_TypedStorage._new_sharedcCs|jjSr)rrrrrrrdsz_TypedStorage._cdatacCs|jjSr)rrrrrrrhsz_TypedStorage.devicecCst|Srrrrrrr'lsz_TypedStorage.sizecCs8zt|jWSty2td|jdYn0dS)Nzdtype z is not recognized)rr)KeyErrorrrrrpickle_storage_typeos z!_TypedStorage.pickle_storage_typecCs&t}tj||ddt|ffSrirjrprrrrrusz_TypedStorage.__reduce__cCs |jSr)rr.rrrrr.zsz_TypedStorage.data_ptrcCs|j||dSr)rr?r,r>rrrr?}sz_TypedStorage.resize_cOstj|i|Sr)rrHr5rrrrHsz_TypedStorage._free_weak_refcOs|jj|i|Sr)rr@rrrrr@sz_TypedStorage._weak_refrcOs|tkrd|durtn|}t|dur,dn|}|jdkrLtd|jtjj|d|i|}nH|dusxt|dkrtd|durtd|j }tjj|d|i|}t||dS)Nr~z4_TypedStorage.from_buffer: Not available for device r)zgfrom_buffer: 'dtype' can only be specified in _UntypedStorage.from_buffer and _TypedStorage.from_bufferzhfrom_buffer: 'device' can only be specified in _UntypedStorage.from_buffer and _TypedStorage.from_bufferr) rrWrrr*rrr6rYr))r1r)rrrrrrrr6s" z_TypedStorage.from_buffercCs^t|tjstdt|tjg|j|jd|| }| | krZ| }|Sr) rrWr)rr*rrrrrr.rcrrrrrs  $z_TypedStorage._tocCs |tjSrrrrrrrsz_TypedStorage.doublecCs |tjSrrrrrrrsz_TypedStorage.floatcCs |tjSrrrrrrrsz_TypedStorage.halfcCs |tjSrrrrrrrsz_TypedStorage.longcCs |tjSrrrrrrrsz_TypedStorage.intcCs |tjSrrrrrrrsz_TypedStorage.shortcCs |tjSrrrrrrrsz_TypedStorage.charcCs |tjSrrrrrrrsz_TypedStorage.bytecCs |tjSrrrrrrrsz_TypedStorage.boolcCs |tjSrrrrrrrsz_TypedStorage.bfloat16cCs |tjSrrrrrrrsz_TypedStorage.complex_doublecCs |tjSrrrrrrrsz_TypedStorage.complex_floatcCsB|tkrtdt|jj|||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) rrrrr from_filerWrrr))r1filenameZsharedr'rrrrrrs  z_TypedStorage.from_filecOst|jjj|i|Sr)rrr_expiredr5rrrrsz_TypedStorage._expiredcCs |jSr)rrArrrrrAsz_TypedStorage.is_pinnedcOs|jj|i|Sr)rr=rrrrr=sz_TypedStorage._write_filecOs|jj|i|Sr)rrBrrrrrBsz_TypedStorage._set_from_filecOs|jj|i|Sr)rrCrrrrrCsz_TypedStorage._set_cdatacOs|jj|i|Sr)rrDrrrrrDsz_TypedStorage._share_cuda_cCs |jSr)rrErrrrrEsz_TypedStorage.is_sharedcOstjj|i|Sr)rWrrFr5rrrrF sz_TypedStorage._new_shared_cudacOs*|jj|i|\}}}||||fSr)rr/r,)rrrZmanager_handleZstorage_handler'rrrr/ sz"_TypedStorage._share_filename_cpu_cCs|j|Sr)rr<rrrrr<s z_TypedStorage._shared_decrefcOstjj|i|SrrWrr:)r1rrrrrr_release_ipc_countersz"_TypedStorage._release_ipc_countercOs|jj|i|Sr)rrGrrrrrGsz_TypedStorage._shared_increfcOs&|jj|i|\}}|||fSr)rr0r,)rrrfdr'rrrr0sz_TypedStorage._share_fd_cpu_cCsh|jtvrdSt|j}|jjdvr,dS|jjdkrsF     F