iFDSrSSKrSSKrSSKJrJrJrJrJr SSK J r SSK J r \"S5r "SS\\ 5r"S S 5r"S S \5r"S S\5r"SS5rS\4SjrS\ 4SjrS\S\\S\4Sjrg)zProvide TOSA specification parsing and context utilities. Use these helpers to parse and validate TOSA profile/extension strings and to manage a lowering-time context for the active specification. N)DictGenericListSetTypeVarVersion)ShapeEnvTcj\rSrSrSrSSS\SS4Sjr\SSS\S4S j5r SSS\\4S jr S r g) TosaSpecMappingc0UlgN_mappingselfs i/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/backends/arm/tosa/specification.py__init__TosaSpecMapping.__init__s :< specTosaSpecificationvaluereturnNctUR(dUR(a[SU35e[U[5(a'[ UR 5S:a[SU35eUR5nX0R;a/URU'URURU5 g)a;Adds a value to the mapping for the given TOSA specification. The specification is normalized to its canonical form, which means that only the version and profiles are considered, without extensions. This allows for grouping of values under the same TOSA specification regardless of the extensions they may have. z2TosaSpecMapping does not support extensions, got: z9TosaSpecMapping does not support multiple profiles, got: N) is_U55_subset extensions ValueError isinstance Tosa_1_00lenprofiles_canonical_keyrappend)rrr norm_specs raddTosaSpecMapping.adds   DTFK  dI & &3t}}+=+AKD6R '') MM )')DMM) $ i ''.rc[U[5(a[[UR5nUSS1:Xa>URn[ R SUS35[ R SUS35/$U/$)NFPINTTOSA-z+FPz+INT)r"r#setr%versionrcreate_from_string)rr%r0s r_get_base_specsTosaSpecMapping._get_base_specs3su dI & &!$T]]!3HD%=(,,%885 9MN%885 9NOv rcURU5n/nUH>nUR5nURURR U/55 M@ [ U5S:Xa[ SU35eU$)zReturns a list of values associated with the given TOSA specification. The specification is normalized to its canonical form, which means that only the version and profiles are considered, without extensions. rz(No values found for TOSA specification: )r2r&extendrgetr$KeyError)rr base_specsresultbase norm_bases rr6TosaSpecMapping.get@st))$/ D++-I MM$--++Ir: ; v;! EdVLM M rr) __name__ __module__ __qualname____firstlineno__rr r) staticmethodrr2r6__static_attributes__rrr r sc=/+/A/$/2 1 d;N6O  +Qrr c\\rSrSr%Sr\\S'\\S'\\ \S'\"S5\"S5/r SS /r S \4S jr S \4S jr S \ S \4SjrS\S\\ 4Sjr\S\\ -S \4Sj5r\S \S4Sj5r\S\ S \S4Sj5r\S\\ -S \S4Sj5r\S\ S S4Sj5rSSjrSrg)rTaRepresent a TOSA specification. A specification includes a semantic version, one or more profiles, and optional extensions and levels (for example ``8k``). The encoded form follows ``TOSA-..+[+][+...]``. Profiles use uppercase (for example ``INT``, ``FP``); levels and extensions use lowercase. Attributes: version (Version): Parsed TOSA semantic version. is_U55_subset (bool): True if the ``u55`` subset is requested. r0rr z1.0z1.1r-r,rc[e)z0Return True if integer operations are supported.NotImplementedErrorrs rsupport_integer!TosaSpecification.support_integeri!!rc[e)z7Return True if floating-point operations are supported.rGrs r support_floatTosaSpecification.support_floatmrKr extensionc[e)Return True if an extension is supported and enabled. Args: extension (str): Extension name (for example ``int4``, ``bf16``). Returns: bool: True if the extension is valid for the active profiles and selected. rG)rrOs rsupport_extension#TosaSpecification.support_extensionqs "!rextrascxXl/UlSU;UlUR(aURS5 gg)zInitialize the base specification. Args: version (Version): Parsed TOSA semantic version. extras (List[str]): Remaining tokens such as profiles, levels, and extensions. u55N)r0r rremove)rr0rTs rrTosaSpecification.__init__}s6 "f_    MM%  rc[U[5(aUnO:[U5nURS5(aU[ S5Sn[U5n[UR SUR 35$)Nr..)r"r str startswithr$majorminor)clsr0parsed version_strs r_normalize_version$TosaSpecification._normalize_versionsf gw ' 'Fg,K%%g..)#g,.9 [)F&,,q788rc /nURH<nURH)nURURSUSU355 M+ M> U$)z5Return specs for all supported versions and profiles.r.+)_SUPPORTED_VERSIONS_SUPPORTED_PROFILESr'r1)r_specsr0profiles rall_versions_and_profiles+TosaSpecification.all_versions_and_profilessT,...G22 S33eG9AgY4OPQ3/ rric UR5nX R;a[SU35eURVs/sHnUR SUSU35PM sn$s snf)z;Return specs for all supported versions of a given profile.zUnsupported TOSA profile: r.re)upperrgr!rfr1)r_ri normalizedr0s rall_versions_for_profile*TosaSpecification.all_versions_for_profiless]]_ 44 49'CD D22 2  " "U7)1ZL#A B2   s!A c RURU5nURVs1sHo3RUR4iM nnURUR4U;a[ SU35eUR Vs/sHnUR SUSU35PM sn$s snfs snf)zDekk!n-G[[^))#.Fv~ #PQUPV!WXX--.VaV$W55Va#G44$';G9F4&%QRRNtfUVVrc[e)zYReturns a new TosaSpecification instance with only version and profiles (no extensions). rGrs rr& TosaSpecification._canonical_keys "!r)r rr0N)rr)r=r>r?r@__doc__r __annotations__boolrr[rfrgrIrMrRr classmethodrbrjrortrAr1r&rBrCrrrrTsH S "5>75>: $-"""t" "3 "4 " ! !c ! 93 97 9 9$/B*C s t^[TU]X5 URnUVs/sHoDUR;PM snR S5S:Xa[ SURSUS35eUVs/sHoDUR;PM snR S5[ UR5:a[ SURSUS35eUVs/sHoDUR;dMUPM snUlURHnURU5 M SU;Ul UR(aURS5 /mURHnTURU- mM [U4SjU55(d![ S UUR5S U35eX l g s snfs snfs snf) aInitialize the 1.00 specification and validate extras. Args: version (Version): Semantic version (major=1, minor=0). extras (List[str]): Tokens including profiles, level, and extensions. Raises: ValueError: If no/too many profiles are provided or extensions are invalid. Trz No profile (z ) found in: rZzToo many profiles (8kc3,># UH oT;v M g7frrC).0ecombined_extensionss r %Tosa_1_00.__init__..$s >t D Is556l6(!L  28 8A'' ' 8 > >t Ds  " "H  %c&<&<%=\&QRS  %+JFq33I3I.IFJ A MM!  == MM$  A 3#7#7#: : %>%@$A*NNrc`[[UR5UR5-5$)zvReturn a stable hash for use in sets and dict keys. Returns: int: Hash value derived from version and profiles. )hashr[r0rrs r__hash__Tosa_1_00.__hash__Bs&C %(A(A(CCDDrotherc[U[5(aAURUR:H=(a! UR5UR5:H$g)zReturn True if another instance represents the same spec. Args: other (object): Object to compare. Returns: bool: True if versions and profiles match. F)r"r#r0r)rrs r__eq__Tosa_1_00.__eq__KsJ eY ' 'LLEMM1))+u/I/I/KK rc SUR;$)z.Return True if the ``INT`` profile is present.r-r%rs rrITosa_1_00.support_integer[s %%rc SUR;$)z-Return True if the ``FP`` profile is present.r,rrs rrMTosa_1_00.support_float_st}}$$rrOcURnURH(nXRU;dMXR;dM( g g)rQTF)rr%rr )rrOr_rs rrRTosa_1_00.support_extensioncs=nnA0033 __8TrcSSKJn U"URRSURRS35n[ X R R55$)zReturns a new Tosa_1_00 instance with only major.minor version and profiles (no extensions). Patch version is set to zero for normalization. rrrZz.0)packaging.versionr r0r]r^r#r%copy)rr norm_versions rr&Tosa_1_00._canonical_keyssL .$,,"4"4!5Qt||7I7I6J"MN }}'9'9';<r?r@rrr[rrrrr rrrrintrobjectrrIrMrRr&rB __classcell__)rs@rr#r#s 3iNS 4@ 0!0!c0!d9c9;; OE#EFt &%34 = =rr#c&\rSrSr/SQ/SQS.rSrg)ri)shapeint64rrrrrV) rrrrrrrrrandommxfpblockscale_ue5m3rrCN)r=r>r?r@rrBrCrrrrsG  " rrc\rSrSr%Sr\R "S5r\R \S'\R "S5r \R \S'S S\ S\ S-4S jjr S r S rS rg)TosaLoweringContextiapManage the TOSA specification context for lowering. For now, only the active ``TosaSpecification`` is tracked, but this can be extended to carry additional lowering policies or configuration. Attributes: tosa_spec_var (contextvars.ContextVar): Context variable storing the active spec. spec (TosaSpecification): Specification passed to the context manager. tosa_spec tosa_spec_var shape_env shape_env_varNrcXlX lg)zInitialize the lowering context with a specification. Args: spec (TosaSpecification): Active specification to put into context. N)rr)rrrs rrTosaLoweringContext.__init__s "rc[RRUR5Ul[R RUR 5UlU$)zmSet the context variable and return self. Returns: TosaLoweringContext: This context manager instance. )rrr/rtosa_spec_tokenrrshape_env_tokenrs r __enter__TosaLoweringContext.__enter__sC 3@@DDTYYO2@@DDT^^T rc[RRUR5 [RRUR 5 g)zReset the context variable to its previous state. Args: exc_type (type | None): Exception type, if any. exc_value (BaseException | None): Exception instance, if any. traceback (TracebackType | None): Traceback, if any. N)rrresetrrr)rexc_type exc_value tracebacks r__exit__TosaLoweringContext.__exit__s8 ))//0D0DE))//0D0DEr)rrrrr)r=r>r?r@r contextvars ContextVarrrrrr rrrrBrCrrrrsb -8,B,B;,OM;))O,7,B,B;,OM;))O#.#8d?#  Frrrcr[RR5$![a [ S5ef=f)zGet the current ``TosaSpecification`` from the lowering context. Returns: TosaSpecification: Active specification retrieved from the context var. Raises: RuntimeError: If called outside a ``TosaLoweringContext``. 6Function must be executed within a TosaLoweringContext)rrr6 LookupError RuntimeErrorrCrrget_context_specrs:U"004466 USTTUs 6c[RR5nUc [S5e[RR5$![a [S5ef=f)zGet the current ShapeEnv from the lowering context, if any. Returns: ShapeEnv: Active ShapeEnv retrieved from the context var. Raises: RuntimeError: If called outside a ``TosaLoweringContext`` or if no ShapeEnv is set. z2No ShapeEnv set in the current TosaLoweringContextr)rrr6rr)rs rget_context_shape_envrs`U'5599;  ST T"004466 USTTUs A A A"rrhcN[RU5nUH nX1;dM g g)aCheck if a specification matches any in a set, considering base specs. Args: spec (TosaSpecification): Specification to check. specs (Set[TosaSpecification]): Set of specifications to match against. Returns: bool: True if a match is found, False otherwise. TF)r r2)rrhr8r:s rtosa_spec_in_setrs+!006J = r)rrr{typingrrrrrrr %torch.fx.experimental.symbolic_shapesr r r rr#rrrrrrrCrrrs  44%: CL;gaj;|L"L"^Z=!Z=z y ,2F2Fj U+ U UxU(,S9J5KPTr