<ЦiA%SrSSKrSSKrSSKJr SSKJrJrJrJ r J r J r J r SSK r SSKr SSK Jr SSKJrJr SSKJr SS KJrJr S q\R4"5r\R8"5r/q\ \ \/S4\ \4\ S '\!"\ RDS S 5r#\ \\\$S4r%\"5q&Sr'\ \ RDRP\ S'S\)4Sjr*\*"5(aC\ RDRVr+\ RDRXr-\ RDR\r/O\"S5r+S\$S\$4Sjr-S\$S\$4Sjr/\"SS9S\$4Sj5r0S\)4Sjr1Sr2Sr3Sr4Sr5Sr6"SS5r7"S S5r"S!S"\5r8S\%SS4S#jr9S;S\ \%S\4S$jjr:\"S5S;S\ \%S\\\44S%jj5r;S;S\ \%S\+4S&jjr"S)S*5r?S+\ S,S\?4S-jr@S.rAS+\4S/jrBS;S\ \%S\4S0jjrCS;S\%SS4S1jjrDS\ \4S2jrES\4S3jrFS\ RS\ RDRP4S4jrGSrsrrdefault_generatorsreturnc6[RR$)z(Return true if compile with XPU support.)torch_C_has_xpurrr _is_compiledr!#s 88  r_XpuDevicePropertiesr c[S5eNz(PyTorch was compiled without XPU supportNotImplementedErrorr s r_exchange_devicer'0!"LMMrc[S5er$r%r s r_maybe_exchange_devicer*3r(r)maxsizec^[5(dg[RR5$)z*Return the number of XPU device available.r)r!rr_xpu_getDeviceCountrrr device_countr.7s! >> 88 ' ' ))rc[5S:$)z7Return a bool indicating if XPU is currently available.r)r.rrr is_availabler0?s >A rcg)zKReturn a bool indicating if the current XPU device supports dtype bfloat16.Trrrris_bf16_supportedr2Es rc:[=(a [5(+$)z8Return whether PyTorch's XPU state has been initialized.) _initialized_is_in_bad_forkrrris_initializedr6Js  1 111rc [5(aU"5 gURSS5(a*[RU[R "55 gURSS5(a*[R U[R "55 g[RU[R "545 g)Nseed_allFseed) r6get_lazy_seed_trackerqueue_seed_all traceback format_stack queue_seedrappend)callablekwargss r _lazy_callrCOs  ::j% ( (  - -h 8N8N8P Q ZZ & &  ) )(I4J4J4L M  (I,B,B,D!E Frc[5 g)zInitialize 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_initrrrinitrF]s Lrc [5(d[[S5(ag[ [5(a SSS5 g[ 5(a [ S5e[ 5(d [S5e[RR5 S[l [RS[R555 [H upU"5 M ['[S5 SqSSS5 g![ a1nS[#U5SSR%U53n[!U5UeSnAff=f!['[S5 f=f!,(df  g=f) Nis_initializingzuCannot re-initialize XPU in forked subprocess. To use XPU with multiprocessing, you must use the 'spawn' start methodz#Torch not compiled with XPU enabledTc36# UHo(dM Uv M g7fNr).0callss r _lazy_init..}sX0NuRWUU0Ns  z5XPU call failed lazily at initialization with error: z' XPU call was originally invoked at: )r6hasattr_tls_initialization_lockr5 RuntimeErrorr!AssertionErrorrr _xpu_initrHrextendr; get_calls Exceptionstrjoindelattrr4) queued_callorig_tracebackemsgs rrErEfs=74):;;          I ~~ !FG G  $X0B0L0L0NXX -/<+ 0M0= D+ , C  2!0OPSTUPVxXBBD''.BYAZ\$C.a/ 0 D+ ,A  sNEB E ED%E)E D?,D::D??EEE E%c>\rSrSrS\4SjrSrS\S\S\4SjrS r g ) _DeviceGuardindexcXlSUlgNidxprev_idx)selfrcs r__init___DeviceGuard.__init__s rc`[RRUR5UlgrJrxpur'rhrirjs r __enter___DeviceGuard.__enter__ 22488< rtypevaluer=c`[RRUR5Ulgrrror*rirhrjrtrur=s r__exit___DeviceGuard.__exit__9933DMMBrrgN) __name__ __module__ __qualname____firstlineno__intrkrqrry__static_attributes__rrrraras-c=SrracB\rSrSrSrS\4SjrSrS\S\S\4SjrS r g ) r zContext-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``. c0[USS9UlSUlg)NToptionalrf)rrhri)rjr s rrkdevice.__init__s$Vd; rc`[RRUR5UlgrJrnrps rrqdevice.__enter__rsrrtrur=c`[RRUR5Ulgrrwrxs rrydevice.__exit__r{rrgN) r|r}r~r__doc__rrkrqryrrrrr r s2s=Src,^\rSrSrSrU4SjrSrU=r$) device_ofa Context-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. cj>UR(aUR5OSn[TU] U5 gre)is_xpu get_devicesuperrk)rjobjrh __class__s rrkdevice_of.__init__s$"%**cnn" rr)r|r}r~rrrkr __classcell__)rs@rrrsrrcz[5 [U5nUS:a [RR U5 gg)zSet 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)rErrr_xpu_setDevicer s r set_devicers1L v &F { 'rc,[U5R$)amGet 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 )get_device_propertiesnamer s rget_device_namers ! ( - --rc[U5n[U5Vs0sH'nURS5(aMU[X5_M) sn$s snf)aGet 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 )__ _pybind11_)rdir startswithgetattr)r propsprops rget_device_capabilityrsR "& )EJ D34 #ge""  s A A c~[5 [USS9nUS:dU[5:a [S5e[ U5$)zGet the properties of a device. 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 TrrzInvalid device index)rErr.rT_get_device_propertiesr s rrrs<L v 5F zV|~-344 !& ))rcR[5 [RR5$)z0Return the index of a currently selected device.)rErr_xpu_getDevicerrrcurrent_devicersL 88 " " $$rc[U[5(a[R"U5nU$[U[5(a[R"SU5nU$)zReturn the torch.device type object from the passed in device. Args: device (torch.device or int or str): selected device. ro) isinstancerYrr rr s r _get_devicer sJ &#f% M FC eV, MrcZ\rSrSr%Sr\S\S'S\S4SjrSrS\ S \ S \ 4S jr S r g ) StreamContextiaContext-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_streamstreamc\Xl[SS5UlURcSUlgg)NTrf)rrrh)rjrs rrkStreamContext.__init__&s+ $T40 88 DH rcURnUbURS:Xag[RR S5UlUR R UR :waL[ UR 5 [RR UR 5UlSSS5 [RRU5 g!,(df  N.=fre) rrhrrocurrent_streamsrc_prev_streamr dst_prev_stream set_stream)rjrs rrqStreamContext.__enter__,s[[  R $yy77=    & &**;*; ; ))*',yy'?'? @Q@Q'R$+ Z(+*s >/C C#rtrur=c0URnUbURS:XagURRUR:wa)[R R UR5 [R R UR5 gre)rrhrr rrorr)rjrtrur=rs rryStreamContext.__exit__8sj[[  R     & &**;*; ; II !5!5 6 T112r)rrhrrN) r|r}r~rrr__annotations__rkrqrryrrrrrrsF +,,x(:; )3S333rrrrc[U5$)zWrap 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``. )rrs rrrCs   rcB[RRUUUS9 g)zset 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_setStreamrs r_set_stream_by_idrLs$ HH!rcrUcg[5 [URURURS9 g)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)rErrrrrs rrrZs4~L""((&&rc[5 [RR[ USS95n[ USUSUSS9$)a:Return 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). Trrrr)rErr_xpu_getCurrentStreamrr)r streamdatas rrrmsJL//&40J Q-jmTU rch[5 [USS9n[RR U5$)aWait 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)rErrr_xpu_synchronizer s r synchronizers*L v 5F 88 $ $V ,,rc[5(d/$[RR5nUc/$UR 5$)z> ++-J    rcl[5n[U5S:XagSSRSU553$)zIReturn XPU AOT(ahead-of-time) build flags this library was compiled with.rrOz-device ,c3$# UHov M g7frJr)rKarchs rrM$get_gencode_flags..s: t s)rlenrZ) arch_lists rget_gencode_flagsrs5I 9~ chh: ::; <>[T5nURT5 grJ)r set_offset)default_generator final_devicers rcb!_set_rng_state_offset..cbs*<8$$V,rN)rrC)rr rrs` @r_set_rng_state_offsetrsv&L-rNrcb[5 [U5n[U5nUR5$)a4Return 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. )rErr get_offset)r rrs r_get_rng_state_offsetrs-Lv&L&|4  ' ' ))r) empty_cachemax_memory_allocatedmax_memory_reserved mem_get_infomemory_allocatedmemory_reserved memory_statsmemory_stats_as_nested_dictreset_accumulated_memory_statsreset_peak_memory_stats) get_rng_stateget_rng_state_all initial_seed manual_seedmanual_seed_allr9r8 set_rng_stateset_rng_state_all)+rrrrrrr rr.rrrrrrrr get_streamrFrr0r2r6rrrrrrrrrrrr9r8rrrrrstreamsrrJ)ro)`r threadingr= functoolsrtypingrrrrrr r rtorch._Cr _device torch._utilsr r_utilsrrrrr4localrQLockrRrrYrrrr5r _device_tr;r Generatorboolr!r"_xpu_exchangeDevicer'_xpu_maybeExchangeDevicer*r.r0r2r6rCrFrErarrrrrrrrrrrrrrrrrrmemoryrrrrrrrrrrrandomrrrrrr9r8rr__all__rrrrsrDDD #6%"  ~~'t (2t8 d3i '(%(($6 F '3T) * %'02E%((,,-2d >> 8888xx33"XX>>''=>NNNNsNsN 1*c**d  2 G%P  (   (y (T ( .HY/ .3 . 4(9"5c3h4*(9"5*AU*"%% c3 45 %,, '3'3T!8./!M! v&8I.&$ - -T -tCy=3= -5<< -EHH,>,> -:? sC56 & *%S%,,(>"? *C *"      , r