<Цi+ %SrSSKrSSKrSSKJrJrJrJrJrJ r J r SSK r SSK J r Jr SSKJrJrJr SSKJr SSKJr \ \ \\4r\ R4r\ R6rS q/q\\ \/S4\\4\S '\R>"5r \RB"5r"\"5r#S r$S r%S \&4Sjr'S)Sjr("SS\)5r*S \&4Sjr+S \&4Sjr,S*S\\S S4Sjjr-S \4Sjr.S \4Sjr/S*S\\S \4Sjjr0S*S\\S \4Sjjr1S*S\\S \ \\44Sjjr2S)Sjr3S\4Sjr4S\S S4Sjr5"SS5r "S S!5r6S\S"S \64S#jr7S+S\ \\\ R4S \4S$jjr8S+S%\S\ \\\ R4S S4S&jjr9SS'K:7 /S(Qr;g),zH This package enables an interface for accessing MTIA backend in python N)AnyCallableDictListOptionalTupleUnion)deviceTensor) _dummy_type_LazySeedTracker classproperty)Device)_get_device_indexF _queued_callsc[5 gN) _lazy_initR/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/mtia/__init__.pyinitr!sLrc:[=(a [5(+$)z9Return whether PyTorch's MTIA state has been initialized.) _initialized_is_in_bad_forkrrris_initializedr%s  1 111rreturnc>[RR5$r)torch_C_mtia_isInBadForkrrrrr*s 88 % % ''rc [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_initializingzwCannot re-initialize MTIA in forked subprocess. To use MTIA with multiprocessing, you must use the 'spawn' start methodzTorch not compiled with MTIA enabled. Ensure you have `import mtia.host_runtime.torch_mtia` in your python src file and include `//mtia/host_runtime/torch_mtia:torch_mtia` as your target dependency!Tc36# UHo(dM Uv M g7frr).0callss r _lazy_init..PsX0NuRWUU0Ns  z6MTIA call failed lazily at initialization with error: z( MTIA call was originally invoked at: )rhasattr_tls_initialization_lockr RuntimeError _is_compiledAssertionErrorr r! _mtia_initr$rextend_lazy_seed_tracker get_calls ExceptionstrjoinDeferredMtiaCallErrordelattrr) queued_callorig_tracebackemsgs rrr.sH74):;;          I ~~ *   $X0B0L0L0NXX -/<+ <M0= D+ , Y  H!<PQTUVQWPXYCCE77>CZB[]04!; < D+ ,W  sNEB E ED%E)E D?,D::D??EEE E%c\rSrSrSrg)r8arN)__name__ __module__ __qualname____firstlineno____static_attributes__rrrr8r8asrr8c>[RR5$)z*Return true if compiled with MTIA support.)r r! _mtia_isBuiltrrrr/r/es 88 ! ! ##rc<[5(dg[5S:$)z'Return true if MTIA device is availableFr)r/ device_countrrr is_availablerIjs >> >A rr c[RRU5 [RR 5sSSS5 $!,(df  g=f)z?Waits for all jobs in all streams on a MTIA device to complete.N)r mtiar r!_mtia_deviceSynchronizer s r synchronizerNrs0   6 "xx//1 # " "s A Ac>[RR5$)z,Return the number of MTIA devices available.)r r!_accelerator_hooks_device_countrrrrHrHxs 88 3 3 55rc>[RR5$)z0Return the index of a currently selected device.)r r!%_accelerator_hooks_get_current_devicerrrcurrent_devicerS}s 88 9 9 ;;rcP[RR[USS95$)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.mtia.current_device`, if :attr:`device` is ``None`` (default). Toptional)r r!_mtia_getCurrentStreamrrMs rcurrent_streamrX! 88 * *+[RR5$)zEmpty the MTIA device cache.)r r!_mtia_emptyCacherrr empty_cacherbs 88 $ $ &&rstreamcJUcg[RRU5 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``. N)r r!_mtia_setCurrentStreamrcs r set_streamrgs~ HH##F+rcf[U5nUS:a [RRU5 gg)zSet the current device. Args: device (torch.device or int): selected device. This function is a no-op if this argument is negative. rN)rr r!%_accelerator_hooks_set_current_devicerMs r set_devicerjs,v &F { 66v>rcB\rSrSrSrS\4SjrSrS\S\S\4SjrS r g ) r zContext-manager that changes the selected device. Args: device (torch.device or int): device index to select. It's a no-op if this argument is a negative integer or ``None``. c0[USS9UlSUlg)NTrU)ridxprev_idx)selfr s r__init__device.__init__s$Vd; rc`[RRUR5Ulgr)r r!(_accelerator_hooks_maybe_exchange_devicerorp)rqs r __enter__device.__enter__sII$((S rtypevalue tracebackc`[RRUR5Ulg)NF)r r!rurpro)rqrxryrzs r__exit__device.__exit__s88DDT]]Sr)rorpN) r@rArBrC__doc__rrrrvr|rDrrrr r s3sTSrcZ\rSrSr%Sr\S\S'S\S4SjrSrS\ S \ S \ 4S jr S r g ) StreamContextaContext-manager that selects a given stream. All MTIA 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.mtia.Stream cur_streamrccSUlXl[SS5Ul[R R 5(dURcSUl[R R 5(dSO[RRS5Ul [R R 5(dSUl g[RRS5Ul g)NTrn) rrcrror jit is_scriptingrKr\src_prev_streamdst_prev_stream)rqrcs rrrStreamContext.__init__s $T40yy%%''xx ..00Dejj6O6OPT6U  ..00D 6;jj6O6OPT6U rcURnUbURS:Xag[RR S5UlUR R UR :waL[ UR 5 [RR UR 5UlSSS5 [RRU5 g!,(df  N.=fNrn) rcror rKrXrr rrg)rqrs rrvStreamContext.__enter__s[[  R $zz88>    & &**;*; ; ))*',zz'@'@ARAR'S$+ j)+*s >/C C#rxryrzc0URnUbURS:XagURRUR:wa)[R R UR5 [R R UR5 gr)rcrorr r rKrgr)rqrxryrzrs rr|StreamContext.__exit__sj[[  R     & &**;*; ; JJ ! !$"6"6 7 d223r)rrrorrcN) r@rArBrCr~r__annotations__rrrvrr|rDrrrrrsF ,--  x(;<   * 4S 4 4 4rrrc[U5$)aWrap 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``. ..Note:: In eager mode stream is of type Stream class while in JIT it doesn't support torch.mtia.stream )rrfs rrcrcs   rc[R"S[SS9 [R"S/[R US9$)zReturns the random number generator state as a ByteTensor. Args: device (torch.device or int, optional): The device to return the RNG state of. Default: ``'mtia'`` (i.e., ``torch.device('mtia')``, the current mtia device). z.get_rng_state is not implemented in torch.mtia stacklevelr)dtyper )warningswarn UserWarningr zerosuint8rMs r get_rng_staters4 MM8 ;;s%++f ==r new_statec8[R"S[SS9 g)aSets the random number generator state. Args: new_state (torch.ByteTensor): The desired state device (torch.device or int, optional): The device to set the RNG state. Default: ``'mtia'`` (i.e., ``torch.device('mtia')``, the current mtia device). z.set_rng_state is not implemented in torch.mtiarrN)rrr)rr s r set_rng_stater,s MM8r)*)rrIrrNrHrSrXr\ memory_statsmax_memory_allocatedr_rbrjrgrcr rr)rNr)rK)rs+DDD +EE% '3# $     t (2t8 d3i '( ~~'%'2 ((0f I $d$ d2+2t2 6c6 << U8I. U& U U8I. U& UX(9"5XsCxX' ,v , ?y ?T ?(7474t!8/0!]! >%S%,, 67 >V > @F$S#u||%;< " r