Ë ¥ãiTãóÆ—UdZddlZddlZddlmZddlmZddlmZm Z m Z ddlZddlZddl mZmZddlmZdd lmZdd lmZmZmZmZmZddlmZmZdaej<«Zej@«Z!ga"e#e$egdfe#e%fe&d <e'ejPdd„«Z)e«a*dZ+e$ejPjXe&d<de-fd„Z.e.«rCejPj^Z/ejPj`Z1ejPjdZ3ned«Z/de4de4fd„Z1de4de4fd„Z3ed¬«de4fd„«Z5de-fd„Z6dDde-de-fd„Z7de-fd„Z8d„Z9dEd„Z:dEd „Z;dEd!„Z<Gd"„d#«Z=Gd$„d«Z>Gd%„d&e>«Z?deddfd'„Z@dFdede%fd(„ZAed«dFdedeBe%effd)„«ZC dFdede/fd*„ZDde4fd+„ZEde4e%zej|zdej|fd,„ZFded-ede-fd.„ZGGd/„d0«ZHd1e d2deHfd3„ZIdEd4„ZJd1eddfd5„ZKdFdedefd6„ZLdFd7e4dedefd8„ZMdFdeddfd9„ZNde#e%fd:„ZOde%fd;„ZPdej|dejPjXfd<„ZQ dGd=e4de4e%zej|zddfd>„ZRdGde4e%zej|zde4fd?„ZSdd@lTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZeddAlfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZoe dBe$e4e4f«ZpgdC¢Zqy)Hzó This package introduces support for the XPU backend, specifically tailored for Intel GPU optimization. This package is lazily initialized, so you can always import it, and use :func:`is_available()` to determine if your system supports XPU. éN)ÚCallable)Ú lru_cache)ÚAnyÚNewTypeÚOptional)Ú_dummy_typeÚ_LazySeedTracker)ÚDeviceé)Ú_get_device_index)ÚgraphÚgraph_pool_handleÚis_current_stream_capturingÚmake_graphed_callablesÚXPUGraph)ÚEventÚStreamFÚ _queued_callsÚ_xpu_isInBadForkcó—y©NF©róúc/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/torch/xpu/__init__.pyúr&srrÚdefault_generatorsÚreturncó6—tjjS)z(Return true if compile with XPU support.)ÚtorchÚ_CÚ_has_xpurrrÚ_is_compiledr"+s€ä8‰8×ÑÐrÚ_XpuDevicePropertiesÚdevicecó—td«‚©Nz(PyTorch was compiled without XPU support©ÚNotImplementedError©r$s rÚ_exchange_devicer*8ó€Ü!Ð"LÓMÐMrcó—td«‚r&r'r)s rÚ_maybe_exchange_devicer-;r+r)ÚmaxsizecóT—t«sytjj«S)z*Return the number of XPU device available.r)r"rr Ú_xpu_getDeviceCountrrrÚdevice_countr1?s€ôŒ>ØÜ8‰8×'Ñ'Ó)Ð)rcó—t«dkDS)z7Return a bool indicating if XPU is currently available.r)r1rrrÚis_availabler3Gs€ô‹>˜AÑÐrÚincluding_emulationcóp—t«sy|xs(tjj«jS)zKReturn a bool indicating if the current XPU device supports dtype bfloat16.F)r3rÚxpuÚget_device_propertiesÚhas_bfloat16_conversions)r4s rÚis_bf16_supportedr9Ms.€äŒ>Øàò FÜ9‰9×*Ñ*Ó,×EÑEðrcóh—t«sytjj«jS)zGReturn a bool indicating if the current XPU device supports dtype tf32.F)r3rr6r7Ú'has_subgroup_matrix_multiply_accumulaterrrÚis_tf32_supportedr<Ws%€äŒ>Øô 9‰9×*Ñ*Ó,×TÑTÐTrcó(—txrt«S)z8Return whether PyTorch's XPU state has been initialized.)Ú_initializedÚ_is_in_bad_forkrrrÚis_initializedr@bs€äÒ1¤Ó 1Ð1Ð1rcóf—t«r|«y|jdd«r)tj|t j ««y|jdd«r)tj |t j ««ytj|t j «f«y)NÚseed_allFÚseed) r@ÚgetÚ_lazy_seed_trackerÚqueue_seed_allÚ tracebackÚformat_stackÚ queue_seedrÚappend)ÚcallableÚkwargss rÚ _lazy_callrMgs{€ÜÔÙ ð:‰:j %Ô(Ü×-Ñ-¨h¼ ×8NÑ8NÓ8PÕQØ Z‰Z˜ Ô &Ü×)Ñ)¨(´I×4JÑ4JÓ4LÕMô × Ñ (¬I×,BÑ,BÓ,DÐ!EÕFrcó—t«y)z×Initialize PyTorch's XPU state. This is a Python API about lazy initialization that avoids initializing XPU until the first time it is accessed. Does nothing if the XPU state is already initialized. N)Ú _lazy_initrrrÚinitrPus €ô…Lrc ó’—t«sttd«ryt5t«r ddd«yt «rtd«‚t «std«‚tjj«dt_tjd„tj«D«« tD] \}} |«Œ t'td«daddd«y#t $r1}dt#|«›ddj%|«›}t!|«|‚d}~wwxYw#t'td«wxYw#1swYyxYw) NÚis_initializingzuCannot re-initialize XPU in forked subprocess. To use XPU with multiprocessing, you must use the 'spawn' start methodz#Torch not compiled with XPU enabledTc3ó&K—|] }|sŒ|–—Œyw©Nr)Ú.0Úcallss rú z_lazy_init..•sèø€ÒX uÒRWœUÑXùs‚Šz5XPU call failed lazily at initialization with error: z' XPU call was originally invoked at: Ú)r@ÚhasattrÚ_tlsÚ_initialization_lockr?ÚRuntimeErrorr"ÚAssertionErrorrr Ú _xpu_initrRrÚextendrEÚ get_callsÚ ExceptionÚstrÚjoinÚdelattrr>)Úqueued_callÚorig_tracebackÚeÚmsgs rrOrO~sA€äÔœ7¤4Ð):Ô;ØÜ ñ!ôÔØ÷ !ð!ôÔÜðIóð ôŒ~Ü Ð!FÓGÐGä ‰×ÑÔð $ŒÔä×ÑÑXÔ0B×0LÑ0LÓ0NÔXÔXð -Ü/<ò 0Ñ+˜^ð0Ù•Mñ 0ô ”DÐ+Ô,Øˆ÷C!ð!øô2!ò0àOÔPSÐTUÓPVÈxðXBØBDÇ'Á'È.ÓBYÐAZð\ðô$ C›.¨aÐ/ûð0ûô ”DÐ+Õ,ú÷A!ð!úsM¢D=·BD=Â9D(ÃC+Ã D(ÃD=Ã+ D%Ã4,D Ä D%Ä%D(Ä(D:Ä:D=Ä=Ecó6—eZdZdeddfd„Zd„Zdededefd „Zy) Ú_DeviceGuardÚindexrNcó —||_d|_y©Néÿÿÿÿ)ÚidxÚprev_idx)Úselfrks rÚ__init__z_DeviceGuard.__init__§s€ØˆŒØˆ rcó`—tjj|j«|_yrT©rr6r*rorp©rqs rÚ __enter__z_DeviceGuard.__enter__«ó€ÜŸ ™ ×2Ñ2°4·8±8Ó<ˆ rÚtypeÚvaluerGcó`—tjj|j«|_yr©rr6r-rpro©rqrxryrGs rÚ__exit__z_DeviceGuard.__exit__®ó€Ü—9‘9×3Ñ3°D·M±MÓBˆŒØr)Ú__name__Ú __module__Ú__qualname__Úintrrrvrr}rrrrjrj¦s4„ð˜cð dóò=ð˜Sð¨ð¸ôrrjcó:—eZdZdZdeddfd„Zd„Zdededefd „Zy) r$zÓContext-manager that changes the selected device. Args: device (torch.device or int or str): device index to select. It's a no-op if this argument is a negative integer or ``None``. rNcó6—t|d¬«|_d|_y)NT©Úoptionalrn)rrorp)rqr$s rrrzdevice.__init__»s€Ü$ V°dÔ;ˆŒØˆ rcó`—tjj|j«|_yrTrtrus rrvzdevice.__enter__¿rwrrxryrGcó`—tjj|j«|_yrr{r|s rr}zdevice.__exit__Âr~r)rr€rÚ__doc__rrrrvr}rrrr$r$³s9„ñð˜sð tóò=ð˜Sð¨ð¸ôrcó$‡—eZdZdZdˆfd„ZˆxZS)Ú device_ofaContext-manager that changes the current device to that of given object. You can use both tensors and storages as arguments. If a given object is not allocated on a XPU, this is a no-op. Args: obj (Tensor or Storage): object allocated on the selected device. có`•—|jr|j«nd}t‰| |«yrm)Úis_xpuÚ get_deviceÚsuperrr)rqÚobjroÚ __class__s €rrrzdevice_of.__init__Ñs$ø€Ø"%§*¢*ˆcn‰nÔ°"ˆÜ ‰Ñ˜Õr©rN)rr€rr‰rrÚ __classcell__)r‘s@rr‹r‹Çsø„ñ÷ñrr‹cóx—t«t|«}|dk\r tjj |«yy)z¨Set the current device. Args: device (torch.device or int or str): selected device. This function is a no-op if this argument is negative. rN)rOrrr Ú_xpu_setDevicer)s rÚ set_devicer–Ös1€ô„LÜ ˜vÓ &€FØ ‚{Ü ‰×Ñ Õ'ðrcó,—t|«jS)aGet the name of a device. Args: device (torch.device or int or str, optional): device for which to return the name. This function is a no-op if this argument is a negative integer. It uses the current device, given by :func:`~torch.xpu.current_device`, if :attr:`device` is ``None`` (default). Returns: str: the name of the device )r7Únamer)s rÚget_device_namer™ãs€ô! Ó(×-Ñ-Ð-rc ó—t|«}tttttd«tttf}t|«Dcic].}|jd«stt||«x}|«r||“Œ0c}Scc}w)aÐGet the xpu capability of a device. Args: device (torch.device or int or str, optional): device for which to return the device capability. This function is a no-op if this argument is a negative integer. It uses the current device, given by :func:`~torch.xpu.current_device`, if :attr:`device` is ``None`` (default). Returns: dict[str, Any]: the xpu capability dictionary of the device NÚ__) r7r‚ÚfloatÚboolrbrxÚlistÚtupleÚdictÚdirÚ startswithÚ isinstanceÚgetattr)r$ÚpropsÚserializable_typesÚkeyrys rÚget_device_capabilityr¨òsv€ô " &Ó)€Eäœu¤d¬C´°d³¼TÄ5Ì$ÐOÐôu“:öàØ~‰~˜dÔ#Ü¤¨°Ó!4Ð4˜Ð7IÔJð ˆU‰ òðùòsÁ3A>cóF—t«t|d¬«}t|«S)aGet the properties of a device. Returns _XpuDeviceProperties containing the following device properties: - ``name`` (str): device name. - ``platform_name`` (str): SYCL platform name. - ``vendor`` (str): device vendor. - ``device_id`` (int): device identifier (product ID). - ``driver_version`` (str): driver version. - ``version`` (str): runtime version. - ``max_compute_units`` (int): number of parallel compute units. - ``gpu_eu_count`` (int): number of EUs (Execution Unit). - ``max_work_group_size``: (int): maximum number of work-items permitted in a work-group. - ``max_num_sub_groups`` (int): maximum number of sub-groups supported in a work-group. - ``sub_group_sizes``: (list[int]): a list of supported sub-group sizes. - ``local_mem_size`` (int): device local memory capacity that can be allocated per work-group in bytes. - ``has_fp16`` (bool): whether float16 dtype is supported. - ``has_fp64`` (bool): whether float64 dtype is supported. - ``has_atomic64`` (bool): whether 64-bit atomic operations are supported. - ``has_bfloat16_conversions`` (bool): whether bfloat16 conversions are supported. - ``has_subgroup_matrix_multiply_accumulate`` (bool): whether DPAS (Dot Product Accumulate Systolic) is supported. - ``has_subgroup_matrix_multiply_accumulate_tensor_float32`` (bool): whether DPAS with tf32 inputs is supported. - ``has_subgroup_2d_block_io`` (bool): whether 2D block I/O for efficient matrix multiplication is supported. - ``total_memory`` (int): device global memory in bytes. - ``gpu_subslice_count`` (int): number of subslice. - ``architecture`` (int): device architecture identifier (experimental). - ``type`` (str): device type, e.g. 'cpu', 'gpu', accelerator', 'host', 'unknown'. - ``uuid`` (Any): device UUID (Universal Unique ID), 16 bytes. Args: device (torch.device or int or str): device for which to return the properties of the device. Returns: _XpuDeviceProperties: the properties of the device Tr…)rOrÚ_get_device_propertiesr)s rr7r7s!€ôJ„LÜ ˜v°Ô 5€FÜ! &Ó)Ð)rcóR—t«tjj«S)z0Return the index of a currently selected device.)rOrr Ú_xpu_getDevicerrrÚcurrent_devicer5s€ä„LÜ8‰8×"Ñ"Ó$Ð$rcó —t|t«rtj|«}|St|t«rtjd|«}|S)zŒReturn the torch.device type object from the passed in device. Args: device (torch.device or int or str): selected device. r6)r£rbrr$r‚r)s rÚ_get_devicer¯;sD€ô&œ#ÔÜ—‘˜fÓ%ˆð€Mô FœCÔ Ü—‘˜e VÓ,ˆØ€MrÚpeercóŠ—t«t|d¬«}t|d¬«}tjj ||«S)a/Query whether a device can access a peer device's memory. Args: device (torch.device or int or str): selected device. peer (torch.device or int or str): peer device to query access to. Returns: bool: ``True`` if ``device`` can access ``peer``, ``False`` otherwise. Tr…)rOrrr Ú_xpu_canDeviceAccessPeer)r$r°s rÚcan_device_access_peerr³Hs8€ô„LÜ ˜v°Ô 5€FÜ˜T¨DÔ1€DÜ8‰8×,Ñ,¨V°TÓ:Ð:rcóR—eZdZUdZeded<dedddfd„Zd„Zd ed edefd„Z y) Ú StreamContextaContext-manager that selects a given stream. All XPU kernels queued within its context will be enqueued on a selected stream. Args: Stream (Stream): selected stream. This manager is a no-op if it's ``None``. .. note:: Streams are per-device. útorch.xpu.StreamÚ cur_streamÚstreamrNcó\—||_tdd«|_|j€d|_yy)NTrn)r¸rro)rqr¸s rrrzStreamContext.__init__fs-€ØˆŒÜ$ T¨4Ó0ˆŒØ8‰8ÐØˆDHðrcó¼—|j}||jdk(rytjj d«|_|j j|jk7rLt |j«5tjj |j«|_ddd«tjj|«y#1swYŒ)xYwrm) r¸rorr6Úcurrent_streamÚsrc_prev_streamr$Údst_prev_streamÚ set_stream)rqr·s rrvzStreamContext.__enter__ls§€Ø—[‘[ˆ ØÐ §¡¨R¢ØÜ$Ÿy™y×7Ñ7¸Ó=ˆÔð×Ñ×&Ñ&¨*×*;Ñ*;Ò;Ü˜ ×)Ñ)Ó*ñ SÜ',§y¡y×'?Ñ'?À ×@QÑ@QÓ'RÔ$÷ Sä ‰ ×Ñ˜ZÕ(÷ Sð SúsÁ;/CÃCrxryrGcó*—|j}||jdk(ry|jj|jk7r)tj j |j«tj j |j«yrm)r¸ror¼r$rr6r¾r½)rqrxryrGr·s rr}zStreamContext.__exit__xsj€Ø—[‘[ˆ ØÐ §¡¨R¢Øð×Ñ×&Ñ&¨*×*;Ñ*;Ò;ÜI‰I× Ñ ×!5Ñ!5Ô6Ü ‰ ×Ñ˜T×1Ñ1Õ2r) rr€rr‰rÚ__annotations__rrrvrr}rrrrµrµXsM…ñ ðÐ+Ñ,Ó,ð˜xÐ(:Ñ;ðÀóò )ð3˜Sð3¨ð3¸ô3rrµr¸r¶có—t|«S)z³Wrap around the Context-manager StreamContext that selects a given stream. Arguments: stream (Stream): selected stream. This manager is a no-op if it's ``None``. )rµ©r¸s rr¸r¸ƒs€ô˜Ó Ð rcóH—tjj|||¬«y)aset stream specified by the stream id, device index and device type Args: stream_id (int): not visible to the user, used to assigned to the specific stream. device_index (int): selected device index. device_type (int): selected device type. ©Ú stream_idÚdevice_indexÚdevice_typeN)rr Ú_xpu_setStreamrÄs rÚ_set_stream_by_idrÉŒs$€ô ‡HH×ÑØØ!Øðõrcóv—|€yt«t|j|j|j¬«y)aSet the current stream. This is a wrapper API to set the stream. Usage of this function is discouraged in favor of the ``stream`` context manager. Args: stream (Stream): selected stream. This function is a no-op if this argument is ``None``. NrÄ)rOrÉrÅrÆrÇrÂs rr¾r¾šs4€ð€~ØÜ„LÜØ×"Ñ"Ø×(Ñ(Ø×&Ñ&örcó˜—t«tjjt |d¬««}t|d|d|d¬«S)aRReturn the currently selected :class:`Stream` for a given device. Args: device (torch.device or int, optional): selected device. Returns the currently selected :class:`Stream` for the current device, given by :func:`~torch.xpu.current_device`, if :attr:`device` is ``None`` (default). Tr…rrérÄ)rOrr Ú_xpu_getCurrentStreamrr)r$Ú streamdatas rr»r»sJ€ô„LÜ—‘×/Ñ/Ü˜&¨4Ô0ó€JôØ˜Q‘-¨j¸©mÈÐTUÉôðrÚdata_ptrcóš—t«tjj|t |d¬««}t|d|d|d¬«S)a;Return a :class:`Stream` from an external SYCL queue. This function is used to wrap SYCL queue created in other libraries in order to facilitate data exchange and multi-library interactions. .. note:: This function doesn't manage the queue life-cycle, it is the user responsibility to keep the referenced queue alive while this returned stream is being used. The different SYCL queue pointers will result in distinct :class:`Stream` objects, even if the SYCL queues they dereference are equivalent. Args: data_ptr(int): Integer representation of the `sycl::queue*` value passed externally. device(torch.device or int, optional): the device where the queue was originally created. It is the user responsibility to ensure the device is specified correctly. Tr…rrrÌrÄ)rOrr Ú_xpu_getStreamFromExternalrr)rÏr$rÎs rÚget_stream_from_externalrÒ¿sM€ô „LÜ—‘×4Ñ4ØÔ# F°TÔ:ó€JôØ˜Q‘-¨j¸©mÈÐTUÉôðrcón—t«t|d¬«}tjj |«S)a*Wait for all kernels in all streams on a XPU device to complete. Args: device (torch.device or int, optional): device for which to synchronize. It uses the current device, given by :func:`~torch.xpu.current_device`, if :attr:`device` is ``None`` (default). Tr…)rOrrr Ú_xpu_synchronizer)s rÚsynchronizerÕØs*€ô„LÜ ˜v°Ô 5€FÜ8‰8×$Ñ$ VÓ,Ð,rcó~—t«sgStjj«}|€gS|j «S)zØˆ Ü—‘×+Ñ+Ó-€JØÐØˆ Ø×ÑÓÐrcój—t«}t|«dk(ryddjd„|D««›S)zIReturn XPU AOT(ahead-of-time) build flags this library was compiled with.rrXz-device ú,c3ó K—|]}|–—ŒywrTr)rUÚarchs rrWz$get_gencode_flags..ôsèø€Ò:¨œtÑ:ùs‚)rÚÚlenrc)Ú arch_lists rÚget_gencode_flagsráïs5€ä“€IÜ ˆ9ƒ~˜ÒØØ c—h‘hÑ:° Ô:Ó:Ð;Ð<Ð•—t‰«}|j‰«yrT)rãÚ set_offset)Údefault_generatorÚfinal_deviceräs €€rÚcbz!_set_rng_state_offset..cbsø€Ü*¨<Ó8ÐØ×$Ñ$ VÕ,rr’)r¯rM)rär$rêrés` @rÚ_set_rng_state_offsetrësù€ô˜vÓ&€Lö-ôˆr…Nrcób—t«t|«}t|«}|j«S)aLReturn the random number generator state offset of the specified GPU. Args: device (torch.device or int, optional): The device to return the RNG state offset of. Default: ``'xpu'`` (i.e., ``torch.device('xpu')``, the current XPU device). .. warning:: This function eagerly initializes XPU. )rOr¯rãÚ get_offset)r$rérès rÚ_get_rng_state_offsetrîs-€ô„LÜ˜vÓ&€LÜ& |Ó4ÐØ×'Ñ'Ó)Ð)r)Úchange_current_allocatorÚempty_cacheÚget_per_process_memory_fractionÚmax_memory_allocatedÚmax_memory_reservedÚmem_get_infoÚmemory_allocatedÚmemory_reservedÚmemory_snapshotÚmemory_statsÚmemory_stats_as_nested_dictÚMemPoolÚreset_accumulated_memory_statsÚreset_peak_memory_statsÚset_per_process_memory_fractionÚuse_mem_poolÚXPUPluggableAllocator) Ú get_rng_stateÚget_rng_state_allÚinitial_seedÚmanual_seedÚmanual_seed_allrCrBÚ set_rng_stateÚset_rng_state_allÚ_POOL_HANDLE)9rrrµrÿrr³rïrr»rr$r‹r1rðrÚr¨r™r7rárñrrrÒr rrPrr3r9rr@r<rrrròrórôrõrör÷rørùrúrþrûrürCrBr–rýrrr¾r¸ÚstreamsrÕ)Tr’rT)r6)rr‰Ú threadingrGÚcollections.abcrÚ functoolsrÚtypingrrrrÚtorch._CÚtorch._utilsrr Útorch.typesr Ú_utilsrÚgraphsr rrrrrrrr>ÚlocalrZÚLockr[rržrŸrbrÀr¤r r?rErÚ Generatorrr"r#Ú_xpu_exchangeDevicer*Ú_xpu_maybeExchangeDevicer-r‚r1r3r9r<r@rMrPrOrjr$r‹r–r™r r¨r7rr¯r³rµr¸rÉr¾r»rÒrÕrÚrárãrërîÚmemoryrïrðrñròrórôrõrör÷rørùrúrûrürýrþrÿÚrandomrrrrrrCrBrrrÚ__all__rrrúrsÎðòóÛÝ$Ýß)Ñ)ãÛß6Ýå%÷õ÷#ð€Ø€y‡Ó€Ø%y—~‘~Ó'ÐððˆtØ ˆ(2t8Ñ ˜d 3™iÐ 'Ñ(ñóñ˜%Ÿ(™(Ð$6¹ ÓF€Ù%Ó'ÐØ02ÐE˜%Ÿ(™(×,Ñ,Ñ-Ó2ðdóñ „>Ø Ÿ8™8×8Ñ8ÐØ—x‘x×3Ñ3ÐØ"ŸX™X×>Ñ>Ññ'Ð'=Ó>ÐðN ðN¨óNðN sðN¨sóNñ1Ôð*cò*óð*ðdóñ¨4ð¸4óðU˜4óUò2ó Góó%÷P ñ ÷ñô(ôð (vð ( $ó (ñ.˜Fð.¨có.ñˆ4ƒñ &ð°D¸¸c¸±Nòóðð2ñ'*Øð'*àó'*ðT%˜ó%ð ˜˜c™ E§L¡LÑ0ð °U·\±\ó ð ; 6ð ;°ð ;¸Dó ;÷ (3ñ(3ðV!8Ð.Ñ/ð!°Mó!óðvð $óñ&˜6ð¨Vóñ$ sð°FðÀfóñ2 -˜ð -¨$ó -ðt˜C‘yóð=˜3ó=ð -˜5Ÿ<™<ð -¨E¯H©H×,>Ñ,>ó -ð5:ñØðØ˜s™ U§\¡\Ñ1ðà óñ& * #¨¡)¨e¯l©lÑ":ð *Àsó *÷"÷÷÷õ÷& ÷ õ ñ~ u¨S°#¨X¡Ó7€ò:r