o h@sUddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z mZmZddlmZddlZddlmZmZddlmZmZmZe rSddlmZdd gZzddlZd ZWn e yld ZdZYnwe!Z"iZ#d e$d <edddZ%GdddZ&ddZ'Gdd d ej(j)e&Z*ddZ+ej,ddZ-ej,ddZ.ej,ddZ/ddZ0d d!Z1d a2d"d#Z3d$d%Z4d8d'd(Z5d)d*Z6d9d-d.Z7Gd/ddZ8ej9e8j:_9e&j;j9e8j;_9e&jGd2d3d3e8e>d4Z?d:d6d7Z@dS);) annotationsN)AnycastOptional TYPE_CHECKINGTypeVarUnion)Self)_to_type)_bool_intStorage)DeviceLikeType TypedStorageUntypedStorageTFzdict[int, threading.RLock]_share_memory_mapT!Union[_StorageBase, TypedStorage])boundcs8eZdZUded<dZded<dZded<ded<d Zd ed <d Zd ed <ddZdddZ ddZ ddZ ddddZ ddd Z dd!d"Zdd#d$Z ddd'd(Z ddd)d*Zddd+d,Zdd-d.Zdd/d0Zdd1d2Zdd3d4Zd5d6Zd7d8Zedd;d<Zedd=d>Zedd?d@Zed d dAddBdCZeddDdEZeddFdGZddHdIZ dJdKZ!ddLdMZ"ddNdOZ#dPdQZ$dRdSZ%dTdUZ&ddVdWZ'eddXdYZ(dZd[Z)ed\d]Z*e+d^d_Z,e+d`daZ-eddbdcZ.eddddeZ/dfdgZ0ddhdiZ1djdkZ2dldmZ3dndoZ4dpdqZ5drdsZ6fdtduZ7dvdwZ8dxdyZ9dzd{Z:d|d}Z;d~dZddZ?ddZ@ddZAddZBddZCddZDddZEddZFddZGddZHddZIddZJddZKddZLddZMddddZNddddZOddZPeddddZQddZRddZSZTS) _StorageBaser_cdataFr is_sparse is_sparse_csrz torch.devicedeviceN_Optional[torch.device] _fake_devicez_Optional[int]_checkpoint_offsetcOsdSNselfargskwargsrra/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/storage.py__init__5z_StorageBase.__init__returnr cCtrNotImplementedErrorr!rrr$__len__8r&z_StorageBase.__len__cCr(rr)r!idxrrr$ __getitem__;r&z_StorageBase.__getitem__cOr(rr)r rrr$ __setitem__>r&z_StorageBase.__setitem__sourcer non_blocking_Optional[_bool]cCr(rr)r!r1r2rrr$copy_Ar&z_StorageBase.copy_rcCr(rr)r+rrr$newDr&z_StorageBase.newcCr(rr)r+rrr$nbytesGr&z_StorageBase.nbytescC|Sr)r7r+rrr$sizeJz_StorageBase.sizedtype_Optional[str]cCs t|||Sr)r )r!r;r2rrr$typeMs z_StorageBase.typecC(|rtd|ntd}|j||dS)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. cudarr2torchrtor!rr2device2rrr$r?Rsz_StorageBase.cudacCr>)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. hpur@rArDrrr$rFcs z_StorageBase.hpucCr(rr)r+rrr$ element_sizerr&z_StorageBase.element_sizecC|jjSr)rindexr+rrr$ get_deviceur:z_StorageBase.get_devicecCr(rr)r+rrr$data_ptrxr&z_StorageBase.data_ptrcCr(rr)r+rrr$ resizable{r&z_StorageBase.resizablecOr(rr)r rrr$_share_filename_cpu_r&z!_StorageBase._share_filename_cpu_cOr(rr)r rrr$_share_fd_cpu_r&z_StorageBase._share_fd_cpu_r9r cCr(rr)clsr9rrr$_new_using_filename_cpuz$_StorageBase._new_using_filename_cpucCr(rr)rOrrr$_new_using_fd_cpurRz_StorageBase._new_using_fd_cpucOr(rr)rPr"r#rrr$ from_bufferrRz_StorageBase.from_buffer)rr;cCr(rr))rPmanagerobjr9rr;rrr$_new_shared_filename_cpus z%_StorageBase._new_shared_filename_cpucOr(rr)rTrrr$_release_ipc_counter_cudarRz&_StorageBase._release_ipc_counter_cudacOr(rr)rTrrr$_new_with_weak_ptrrRz_StorageBase._new_with_weak_ptrcCr(rr)r+rrr$_shared_decrefr&z_StorageBase._shared_decrefcOr(rr)r rrr$ _write_filer&z_StorageBase._write_filecCr(rr)r!r9rrr$resize_r&z_StorageBase.resize_cOr(rr)r rrr$ _weak_refr&z_StorageBase._weak_refcOr(rr)r rrr$_set_from_filer&z_StorageBase._set_from_filecOr(rr)r rrr$ _set_cdatar&z_StorageBase._set_cdatacOr(rr)r rrr$ _share_cuda_r&z_StorageBase._share_cuda_cCr(rr)r+rrr$ is_sharedr&z_StorageBase.is_sharedcOr(rr)rTrrr$_new_shared_cudarRz_StorageBase._new_shared_cudacOr(rr)r rrr$_shared_increfr&z_StorageBase._shared_increfcOr(rr)rTrrr$_free_weak_refrRz_StorageBase._free_weak_refcCr(rr)r+rrr$is_cudarRz_StorageBase.is_cudacCr(rr)r+rrr$is_hpurRz_StorageBase.is_hpucCr(rr))rPfilenamesharedr7rrr$ from_filerRz_StorageBase.from_filecOr(rr)rTrrr$_expiredrRz_StorageBase._expiredcOr(rr)r rrr$ _byteswapr&z_StorageBase._byteswapcOr(rr)r rrr$ _get_filenamer&z_StorageBase._get_filenamecshdtdjdtd}jjdkrd|Sddfd d tD}|d |S) N[z(device= ) of size ]meta...   c3|] }t|VqdSrstr.0ir+rr$ z(_StorageBase.__repr__.. )rBtypenamerlenr=joinranger9r!info_strdata_strrr+r$__repr__s $ $ z_StorageBase.__repr__cstfddtDS)Nc3|]}|VqdSrrryr+rr$r|z(_StorageBase.__iter__..)iterrr9r+rr+r$__iter__sz_StorageBase.__iter__cCr8r)cloner+rrr$__copy__r:z_StorageBase.__copy__cCs6|di}|j|vr||jS|}|||j<|S)NrB) setdefaultrr)r!memo new_storagerrr$ __deepcopy__s    z_StorageBase.__deepcopy__cC&t}tj||ddt|ffSNF)_use_new_zipfile_serializationioBytesIOrBsave_load_from_bytesgetvaluer!brrr$ __reduce__z_StorageBase.__reduce__cst|Sr)super __sizeof__r9r+ __class__rr$rz_StorageBase.__sizeof__cCst|||jd|S)Return a copy of this storage.r)r=r7rr5r+rrr$rsz_StorageBase.clonecCst|Sz6Return a list containing the elements of this storage.)listr+rrr$tolistz_StorageBase.tolistcCs&|jjdkrt||dS|S)AReturn a CPU copy of this storage if it's not already on the CPU.cpuFrr=rBrr9r5r+rrr$rs z_StorageBase.cpucCs*|jjdkrtj|dd|dS|S)zAReturn a MPS copy of this storage if it's not already on the MPS.mpsrFrr+rrr$r s z_StorageBase.mpscCsdt|tjstdt|tjgtj|jdt t | | }| | kr0|}|SNz*Argument 'dtype' must be torch.dtype, not r;r) isinstancerBr; TypeErrorr=tensoruint8rset_rrrC_typed_storagerKrr!r;storagerrr$r s  z_StorageBase._tor2rcCs"t|tjs t|}t|||Sr)rrBrr )r!rr2rrr$rCs   z_StorageBase.tocC |tjSz"Casts this storage to double type.)r rBdoubler+rrr$r! z_StorageBase.doublecCrz!Casts this storage to float type.)r rBfloatr+rrr$r%rz_StorageBase.floatcCrz Casts this storage to half type.)r rBhalfr+rrr$r)rz_StorageBase.halfcCrz Casts this storage to long type.)r rBlongr+rrr$r-rz_StorageBase.longcCrzCasts this storage to int type.)r rBintr+rrr$r1rz_StorageBase.intcCrz!Casts this storage to short type.)r rBshortr+rrr$r5rz_StorageBase.shortcCrz Casts this storage to char type.)r rBint8r+rrr$char9rz_StorageBase.charcCrz Casts this storage to byte type.)r rBrr+rrr$byte=rz_StorageBase.bytecCrz Casts this storage to bool type.)r rBboolr+rrr$rArz_StorageBase.boolcCrz$Casts this storage to bfloat16 type.)r rBbfloat16r+rrr$rErz_StorageBase.bfloat16cCrz*Casts this storage to complex double type.)r rBcdoubler+rrr$complex_doubleIrz_StorageBase.complex_doublecCrz)Casts this storage to complex float type.)r rBcfloatr+rrr$ complex_floatMrz_StorageBase.complex_floatcCrz&Casts this storage to float8_e5m2 type)r rB float8_e5m2r+rrr$rQrz_StorageBase.float8_e5m2cCrz(Casts this storage to float8_e4m3fn type)r rB float8_e4m3fnr+rrr$rUrz_StorageBase.float8_e4m3fncCrz*Casts this storage to float8_e5m2fnuz type)r rBfloat8_e5m2fnuzr+rrr$rYrz_StorageBase.float8_e5m2fnuzcCrz*Casts this storage to float8_e4m3fnuz type)r rBfloat8_e4m3fnuzr+rrr$r]rz_StorageBase.float8_e4m3fnuzr?Union[str, torch.device]cCs&tjgtj|jdtt||S)a-Determine whether the CPU storage is already pinned on device. Args: device (str or torch.device): The device to pin memory on (default: ``'cuda'``). This argument is discouraged and subject to deprecated. Returns: A boolean variable. r)rBrrrrrr is_pinnedr!rrrr$ras z_StorageBase.is_pinnedcCsN|jjdkrtd|dtjgtj|jdtt| |}| S)a3Copy 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'``). This argument is discouraged and subject to deprecated. Returns: A pinned CPU storage. rz cannot pin 'z' only CPU memory can be pinnedr) rr=rrBrrrrr pin_memoryuntyped_storage)r!r pinned_tensorrrr$rqs  z_StorageBase.pin_memorycCsJddlm}|jjdtjfvr |S|dkr||S||S).See :meth:`torch.UntypedStorage.share_memory_`rget_sharing_strategyr? file_system) torch.multiprocessingrrr=rB_C_get_privateuse1_backend_namerMrN)r!rrrr$ share_memory_s  z_StorageBase.share_memory_rrcCsVddlm}t|}|jdtjdfvr|||dS|dkr&||S||S)>Create a new storage in shared memory with the same data type.rrr?rFrr) rrrBrr=rrrQrS)rPr9rrrrr$ _new_shareds      z_StorageBase._new_sharedcCs|Srrr+rrr$untypedr&z_StorageBase.untypedcCs2tj|}|jrtt|dd}||dS)zSwap bytes in underlying data.N)rB_utils _element_size is_complexmaxrrm)r!r; elem_sizerrr$byteswaps z_StorageBase.byteswapr'r r)r1rr2r3r'r)r'rNF)r;r<r2r r'r)r'r )r9r r'r r'r )r9r r'r<)rrr2r r?rr)U__name__ __module__ __qualname____annotations__rrrrr%r,r/r0r5r6r7r9r=r?rFrGrJrKrLrMrN classmethodrQrSrUrXrYrZr[r\r^r_r`rarbrcrdrerfpropertyrgrhrkrlrmrnrrrrrrrrrrr rCrrrrrrrrrrrrrrrrrrrrrr __classcell__rrrr$r)s                                      rcstfdd}|S)NcsBd}d}t"|j}|tvrt|}ntt|<t||}Wdn1s+wY|durF|Wdn1sAwYz3|g|Ri|W|dury|j|ks]Jtt|t|=WdS1sswYSS|dur|j|ksJtt|t|=Wdw1swYwr)_share_memory_lockrr threadingRLockacquirerelease)r!r"r#to_freeto_waitkeyfnrr$wrappers:       z-_share_memory_lock_protected..wrapper) functoolswraps)r r rr r$_share_memory_lock_protectedsrcsreZdZfddZeddZeddZedd d Zefd d Z efd dZ efddZ Z S)rcs&|jjdkr tdtj|i|S)Nrr$Not available for 'meta' device type)rr=r*rr/r rrr$r/s zUntypedStorage.__getitem__cC |jjdkSNr?rr=r+rrr$rgrzUntypedStorage.is_cudacCrNrFrr+rrr$rhrzUntypedStorage.is_hpur'r<cCr8)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. )rnr+rrr$riszUntypedStorage.filenamectj|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`` )rrr rrr$rs!zUntypedStorage.share_memory_crr)rrNr rrr$rNzUntypedStorage._share_fd_cpu_crr)rrMr rrr$rM rz#UntypedStorage._share_filename_cpu_r) rrrr/rrgrhrirrrNrMrrrrr$rs    "cCstjt|ddS)NF) weights_only)rBloadrr)rrrr$rrcCs<tjtjtjtjtjtjtjtjtj tj tj tj tj tjhSr)rBrrrrfloat8_e8m0fnubits8bits16bits1x8bits2x4bits4x2 complex32uint16uint32uint64rrrr$ _new_dtypessr%cCsitjdtjdtjdtjdtjdtjdtjdtjdtj d tj d tj d tj d tj d tjdtjdtjdtjdS)N DoubleStorage FloatStorage HalfStorage LongStorage IntStorage ShortStorage CharStorage ByteStorage BoolStorageBFloat16StorageComplexDoubleStorageComplexFloatStorage QInt8Storage QInt32Storage QUInt8StorageQUInt4x2StorageQUInt2x4Storage)rBrrrrrint16rrrrrrqint8qint32quint8quint4x2quint2x4rrrr$_dtype_to_storage_type_map+sF     r=cCsddtD}|S)NcSsi|]\}}||qSrr)rzr valrrr$ Isz._storage_type_to_dtype_map..)r=items) dtype_maprrr$_storage_type_to_dtype_mapGsrBc Csx|tjtjtjtjtjfvr/tjtjtjtjtjtjtjtjtjtji}tj ||||d}ntj |||d}| j S)Nr) rBr:r;r<r9r8rint32rrr_untyped_storage)sequencer;rinterpret_dtypes tmp_tensorrrr$_get_storage_from_sequenceMs$  rHcCstr t|ttjfSt|tSr) HAS_NUMPYrrnpinteger)xrrr$_isintfs rMcCstSr)"_always_warn_typed_storage_removalrrrr$&_get_always_warn_typed_storage_removalpr&rOcCst|tsJ|adSr)rrrN) always_warnrrr$&_set_always_warn_typed_storage_removaltsrQrcCs>dd}ts |rd}tj|t|dddtjd<dSdS)NcSsttdsdStjd S)N has_warnedT)hasattr_warn_typed_storage_removal__dict__rrrr$ is_first_time}s  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) stacklevelTrR)rOwarningswarn UserWarningrTrU)rWrVmessagerrr$rTzs rTcCsdtjd<dS)NFrR)rTrUrrrr$!_reset_warn_typed_storage_removalr\modulerxcCs,|ddd}|dtjdfvr|SdS)N.rr?rFr)rsplitrBrr)r^ last_partrrr$_get_device_from_modulesrccseZdZUdZded<dZded<ded<ed d Zedd dZddZ dddddfdd Z dddddddZ eddZ eddZ ddZdddZd d!Zdd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zddd0d1Zd2d3Zd4d5Z  ddd8d9Zddd:d;ZdddddAdBZdCdDZdEdFZddHdIZdJdKZ dLdMZ!dNdOZ"dPdQZ#dRdSZ$dTdUZ%fdVdWZ&dXdYZ'dZd[Z(d\d]Z)ddd`daZ*dddbdcZ+dddeZ,dfdgZ-ddhdidjZ.edkdlZ/edmdnZ0dodpZ1dqdrZ2dsdtZ3dudvZ4dwdxZ5dydzZ6d{d|Z7d}d~Z8ddZ9ddZ:e;ddZe;dddddZ?ddZ@ddZAddZBddZCddZDddZEddZFddZGddZHddZIddZJddZKddZLddZMddZNddZOddZPe;ddZQe;ddZRddZSddZTddZUddZVddZWddZXe;ddZYddZZddÄZ[e;ddhddńZ\ddDŽZ]ddɄZ^dd˄Z_Z`S)rFr rNrr torch.dtyper;cCs|jSr)r;r+rrr$_dtypeszTypedStorage._dtyper'r<cCrH)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.)rDrir+rrr$riszTypedStorage.filenamecCs t|td|||S)Nr)rT_setitemslice_size)r!valuerrr$fill_szTypedStorage.fill_) wrap_storager;r _internalcsb|st|tjjkrtd|tkrt|S|d}|dur(t|d|dur2t|d|durqt|dkrBt|dt|dkrdt |dsdt |dt j j sdt|dt|dt||jt|jd d St|dkr}t|d t |tjst|d t|t|j}|jj|krt|d |d|jjt|||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)r;rrlB 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 rkr;rl)rTrBr_LegacyStorage RuntimeErrorrr__new__rrMr collectionsabcSequencerr=rercrrrr;)rPrkr;rrlr" arg_error_msg cls_devicerrr$rts           zTypedStorage.__new__)rr;rkrlcGs|std}|durUt|dkrt|d|dur!t|dt|tjs2t|dt||dur.r~) rTrBrr;rrr=rrr9rrr+r$__str__Ps $ zTypedStorage.__str__cCtt|Sr)rTrxr+rrr$r\rzTypedStorage.__repr__cs$ttfddtDS)Nc3rrrryr+rr$r|brz(TypedStorage.__iter__..)rTrrr9r+rr+r$r`szTypedStorage.__iter__cCst|t|jSr)rTrcopyrDr+rrr$rdzTypedStorage.__copy__cCrr)rT _deepcopyr!rrrr$rhrzTypedStorage.__deepcopy__cCs|t|j|Sr)rrdeepcopyrDrrrr$rmrzTypedStorage._deepcopycstt|Sr)rTrrr7r+rrr$rprzTypedStorage.__sizeof__cCt||jS)r)rTrrDrr+rrr$rtzTypedStorage.clonecCrr)rTrr+rrr$ryzTypedStorage.tolistcCr)r)rTrrDrr+rrr$r~rzTypedStorage.cpur?rcCst|j|S)a2Determine whether the CPU TypedStorage is already pinned on device. Args: device (str or torch.device): The device to pin memory on (default: ``'cuda'``). This argument is discouraged and subject to deprecated. Returns: A boolean variable. )rTrDrrrrr$rs zTypedStorage.is_pinnedcCst||jj|dS)a8Copy 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'``). This argument is discouraged and subject to deprecated. Returns: A pinned CPU storage. r)rTrrDrrrrr$rs  zTypedStorage.pin_memorycCr)r)rT_share_memory_r+rrr$rrzTypedStorage.share_memory_cC|j|Sr)rDrr+rrr$r zTypedStorage._share_memory_rcCs>|durd}t|}tjj|||d}t||jddS)rNrrTrq)rBrrrrrr;)r!r9rrrrr$rs  zTypedStorage._new_sharedcCrHr)rDrr+rrr$rrzTypedStorage._cdatacCst|jjSr)rTrDrr+rrr$rrzTypedStorage.devicecCrrrr+rrr$r9rzTypedStorage.sizecCs|j|Sr)rDr7rr+rrr$rhszTypedStorage._sizecCrr)rT_pickle_storage_typer+rrr$pickle_storage_typerz TypedStorage.pickle_storage_typec Cs<zt|jWSty}z td|jd|d}~ww)Nzdtype z is not recognized)r=r;KeyError)r!errr$rs z!TypedStorage._pickle_storage_typecCrrrrrrr$rrzTypedStorage.__reduce__cCrr)rT _data_ptrr+rrr$rKrzTypedStorage.data_ptrcCrr)rDrKr+rrr$rrzTypedStorage._data_ptrcCrr)rTrDrLr+rrr$rLrzTypedStorage.resizablecCst||dSr)rT_resize_r]rrr$r^szTypedStorage.resize_cCs|j||dSr)rDr^rr]rrr$rszTypedStorage._resize_cOtj|i|Sr)rrfrTrrr$rfzTypedStorage._free_weak_refcO|jj|i|Sr)rDr_r rrr$r_rzTypedStorage._weak_refcOst|j|i|Sr)rT _from_bufferrTrrr$rUrzTypedStorage.from_bufferrcOs|tkr2|dur tn|}t|durdn|}|jdkr&td|jtjj|d|i|}n$|dus  zTypedStorage._from_buffercCs^t|tjstdt|tjg|j|jd|| }| | kr-| }|Sr) rrBr;rr=rrrrCrrKrrrrr$r s zTypedStorage._tocCt|tjSr)rTr rBrr+rrr$r% zTypedStorage.doublecCrr)rTr rBrr+rrr$r*rzTypedStorage.floatcCrr)rTr rBrr+rrr$r/rzTypedStorage.halfcCrr)rTr rBrr+rrr$r4rzTypedStorage.longcCrr)rTr rBrr+rrr$r9rzTypedStorage.intcCrr)rTr rBrr+rrr$r>rzTypedStorage.shortcCrr)rTr rBrr+rrr$rCrzTypedStorage.charcCrr)rTr rBrr+rrr$rHrzTypedStorage.bytecCrr)rTr rBrr+rrr$rMrzTypedStorage.boolcCrr)rTr rBrr+rrr$rRrzTypedStorage.bfloat16cCrr)rTr rBrr+rrr$rWrzTypedStorage.complex_doublecCrr)rTr rBrr+rrr$r\rzTypedStorage.complex_floatcCrr)rTr rBrr+rrr$rarzTypedStorage.float8_e5m2cCrr)rTr rBrr+rrr$rfrzTypedStorage.float8_e4m3fncCrr)rTr rBrr+rrr$rkrzTypedStorage.float8_e5m2fnuzcCrr)rTr rBrr+rrr$rprzTypedStorage.float8_e4m3fnuzcCs@t|tkr tdt|||tj|j}||d}|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) rTrrsrrkrBrrr;)rPrirjr9rrrrr$rkus zTypedStorage.from_filecOrr)rrlrTrrr$rlrzTypedStorage._expiredcOrr)rDr\r rrr$r\rzTypedStorage._write_filecOrr)rDr`r rrr$r`rzTypedStorage._set_from_filecOrr)rDrar rrr$rarzTypedStorage._set_cdatacOrr)rDrbr rrr$rbrzTypedStorage._share_cuda_cCrr)rT _is_sharedr+rrr$rcrzTypedStorage.is_sharedcCrr)rDrcr+rrr$rrzTypedStorage._is_sharedcOtjj|i|Sr)rBrrdrTrrr$rdrzTypedStorage._new_shared_cudacOs*|jj|i|\}}}||||fSr)rDrMr)r!r"r#manager_handlestorage_handler9rrr$rMs z!TypedStorage._share_filename_cpu_cCrr)rDr[r+rrr$r[rzTypedStorage._shared_decrefcOstjj|i|SrrBrrY)rPrr"r#rrr$_release_ipc_counterrz!TypedStorage._release_ipc_countercOrr)rDrer rrr$rerzTypedStorage._shared_increfcOs&|jj|i|\}}|||fSr)rDrNr)r!r"r#fdr9rrr$rNszTypedStorage._share_fd_cpu_cCsz|jtvrdSt|j}|jjdddtjfvrdS|jjdkr%tntt|jj}zt||WSty<YdSw)Nrr?rF) r;r=rr=rBrrgetattrAttributeError)r! storage_namer^rrr$rs"    z&TypedStorage._get_legacy_storage_classrr)Fr)r1rr2rr)r;r<r2rr'r)rrr2rr'r rrr)arrrrrrrrerirjrtr%rgrhrrr,rr0rfr/rr5r7rr=r?rFrCrGrrJrrrrrrrrrrrrrrrrrr9rhrrrrKrrLr^rrrfr_rUrr rrrrrrrrrrrrrrrrrkrlr\r`rarbrcrrdrMr[rrerNrrrrrr$rs    _ [    # ,                c@seZdZUded<ddZdS)_LegacyStorageMetardr;cCs2t|tkrt|j}||jjko|j|jkSdSr)r=rrcrrr;)rPinstanceryrrr$__instancecheck__s    z$_LegacyStorageMeta.__instancecheck__N)rrrrrrrrr$rs  rc@s0eZdZeddZeddZeddZdS)rrcCs tj||}||dS)rr)rBrrr)rPr9rrrr$rs z_LegacyStorage._new_sharedcOrrrrTrrr$rrz#_LegacyStorage._release_ipc_countercCs(|tj|j}|tj|||dS)Nr)rBrrr;rrX)rPrVrWr9 bytes_sizerrr$_new_shared_filenames z#_LegacyStorage._new_shared_filenameN)rrrrrrrrrrr$rrs  rr) metaclassrc Cs8zt|WSty}z td|d|d}~ww)Nzpickle storage type "z" is not recognized)rBr)rrrrr$#_get_dtype_from_pickle_storage_types  r)r)r^rx)rrx)A __future__rrurrrrrXtypingrrr _Optionalrrrtyping_extensionsr rB torch._utilsr r torch.typesr r rtorch._prims_commonr__all__numpyrJrIModuleNotFoundErrorLockrrrrrrr StorageBaserrcacher%r=rBrHrMrNrOrQrTr\rcr__doc__r=r?rFrCrrrrrrrr$sx      $C      I