il' lSrSSKrSSKJr SSKJrJr SSKrSSKr SSK J r Sr \R\R\R\R \R"\R$\R&\R(\R*4 r"SS\5rS \R0S \4S jrS r"S S5rg)zProvide PyTorch-to-TOSA mapping helpers. Use these utilities to translate PyTorch dtypes and FX node metadata into the TOSA serializer types and shapes used during initial compilation. N)Enum)AnySequence)TosaSpecificationtosa_tensor_namec\rSrSrSr\R Rr\R Rr\R Rr S\R 4Sjr \ S\ 4Sj5r SrSrSrg ) TosaSpecialDtype#z6Special TOSA dtypes not natively expressed in PyTorch.returncUR$)zxReturn the underlying ``ts.DType`` enumerant. Returns: ts.DType: Serializer dtype associated with the enum entry. )valueselfs c/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/backends/arm/tosa/mapping.pyget_tosa_dtypeTosaSpecialDtype.get_tosa_dtype*szzcg)zReturn the FX ``meta`` key that stores special dtypes. Returns: str: Metadata key used to encode :class:`TosaSpecialDtype`. tosa_special_dtyperrrmeta_keyTosaSpecialDtype.meta_key3s$rchU=UR:Xa gUR:Xag[SUS35e)NlUnrecognized TosaSpecialDtype .INT4INT48 ValueErrorrs rmaxTosaSpecialDtype.max=s3  #A$q!IJJrchU=UR:Xa gUR:Xag[SUS35e)Nilrrrrs rminTosaSpecialDtype.minFs3 #A$q!IJJrrN)__name__ __module__ __qualname____firstlineno____doc__tsDTyperrSHAPEr staticmethodstrrr!r$__static_attributes__rrrr r #sc@ HHNNE 88==D HHNNE$c$$KKrr data_typer cU[;a[SU35e[R[R R [R[R R [R[R R[R[R R[R[R R[R[R R[R[R R [R"[R R [R$[R R&[R([R R&[R*[R R,0 nX;a[SU35eX$)zMap a ``torch.dtype`` to a ``ts.DType``. Args: data_type (torch.dtype): PyTorch dtype to convert. Returns: ts.DType: Matching serializer dtype. Raises: ValueError: If the dtype is unsupported or unknown. zUnsupported type: zUnknown type: )UNSUPPORTED_DTYPESr torchfloat32r+r,FP32floatfloat16FP16halfbfloat16BF16int8INT8int16INT16shortint32INT32intboolBOOL)r1 dtype_maps r map_dtyperHPs&&-i[9::  rxx}} RXX]] rxx}} BHHMM   BHHMM RXX^^ RXX^^ RXX^^ 288>> BHHMM I!>)566  rcUR[R55nU[R:Xa/[ US5n[ R RU4US4$URS5c [S5eUSn[U5[LaUSn[U[RRR5(d[SUR35e[!UR"5n[UR%55nURS5bUSnO[['[ U555nXEU4$)aWExtract dtype, shape, and dimension order from FX metadata. Args: meta (dict): FX node ``meta`` containing a ``val`` FakeTensor (or tuple). Returns: tuple[ts.DType, tuple[int, ...], tuple[int, ...]]: Tuple containing tensor dtype, shape, and dimension order. Raises: ValueError: If ``meta['val']`` is not a ``FakeTensor``. valtosa_dim_orderz3Expected node.meta['val'] to be set to a FakeTensorrz?Expected first value in node.meta['val'] to be FakeTensor, got )getr rr-lenr+r,r typetuple isinstancer4 _subclasses fake_tensor FakeTensor __class__rHdtypesizerange)meta special_dtype shape_lenrJrUshape dim_orders rextract_tensor_metar]us!HH-6689M(...U $  d3C.DEE xxNOO u+C CyE!f c5,,88CC D DMcmm_ ]   cii E #((* E xx !-)* %E +, ) $$rc\rSrSrSrS\R R4SjrSr S\ \ -4Sjr S\ S\4S jrS\S\ SS 4S jrS rS rg )TosaArgaCapture and normalize TOSA operator arguments. Use this to convert FX nodes, sequences, and numeric literals into a consistent structure suitable for TOSA serialization. Attributes: name (str): Node name when argument is a ``torch.fx.Node``; empty otherwise. dtype (ts.DType | None): Inferred dtype when available. shape (tuple[int, ...] | None): Inferred shape when available. dim_order (tuple[int, ...] | None): Dimension order, defaulting to ``range(len(shape))``. special (list | None): Captured list when the argument is a sequence. number (float | int | None): Captured numeric value when provided. multiple_output_name (list[str]): Output node names when node has multiple outputs; empty otherwise. argumentcfURR[S5nURU-UlSUR;am[ UR5uo0lUlURR[R5S5=n(aUR5nX0l [UR5n[U5S:aP[SU55(a9UVs/sHofRU-PM snUlURSUlg/Ulgs snf)ztParse a ``torch.fx.Node`` and populate tensor attributes. Args: argument (torch.fx.Node): FX node to inspect. rJNrc3Z# UH!oR[R:Hv M# g7f)N)targetoperatorgetitem).0users r )TosaArg.__process_node..s!Ted++1A1A"Aes)+)rXrLTOSA_TENSOR_NAME_METAnamer]r[r\r rrrUlistusersrMallmultiple_output_names)rrasuffix output_dtype special_typeroris r__process_nodeTosaArg.__process_nodes""#8"=MMF* HMM !7J 8 4L*dn (}}001A1J1J1LdSS|S+::< %J X^^$ u:>c!Te!TTTOT/Uut F0Bu/UD &2215DI)+D &0Vs1D.c$[U5Ulg)zeCapture a sequence argument as ``special``. Args: argument (Sequence[Any]): Sequence to store. N)rnspecialrras r__process_listTosaArg.__process_lists"(^ rcXlg)z]Capture a numeric argument as ``number``. Args: argument (float | int): Numeric value. N)numberrys r__process_numberTosaArg.__process_numbers $, r tosa_specr cf[USS5=[RR:Xa UR 5(dgg=[RR :Xa UR S5(dgg[RR:XaUR S5(dggg)NrUFint4bf16T)getattrr+r,r6 support_floatrsupport_extensionr<)rrs r __validateTosaArg.__validatesdGT* ..00 1 226:: ;   226:: ;rNc<[U[RR5(aFUR U5 UR U5(d[ USUR535eg[U[5(aURU5 g[U[[45(aURU5 g[U[R5(agUcSUlSUl SUlSUlg[#SUS[%U535e)afInitialize the argument wrapper and populate fields. Args: argument (Any): One of ``torch.fx.Node``, ``Sequence``, ``int``, ``float``, ``torch.dtype``, or ``None``. tosa_spec (Optional[TosaSpecification]): Active specification; required for metadata extraction. Raises: RuntimeError: If ``argument`` is of an unsupported type. z doesn't support tensor NrczUnhandled node input argument: z , of type )rPr4fxNode_TosaArg__process_node_TosaArg__validater __repr__r_TosaArg__process_listrDr7_TosaArg__process_numberrUrmr[r\ RuntimeErrorrN)rrars r__init__TosaArg.__init__s h . .    )??9--  k!9$--/9JK  h ) )    )  he - -  ! !( +  h , ,   DIDJDJ!DN -hZz$x.AQ R  rc/n[US5(aURbURSUR<35 URb/URS[R UR35 UR bURSUR <35 URbURSUR<35 [US5(a,URbURSUR<35 [US5(a,URbURS UR<35 [US 5(a8[UR5S :aURS UR<35 URRS SRU5S3$)znReturn a compact representation of populated attributes. Returns: str: Readable list of set attributes. rmzname=zdtype=zshape=z dim_order=rxzspecial=r}znumber=rqrznames=(z, ))hasattrrmappendrUr+ DTypeNamesr[r\rxr}rMrqrTr&join)rattrss rrTosaArg.__repr__sc 4 yy$ uTYYM23zz% vbmmDJJ&?%@ABzz% vdjj^45~~) z$..);<= 4 # # (@ LL8DLL#34 5 4 " "t{{'> LL74;;/2 3 40 1 14--.2 vd&@&@%CDE..))*!DIIe,<+=Q??r)r\rUrqrmr}r[rx)r&r'r(r)r*r4rrrrr7rDrrrErrrrr0rrrr_r_sf$,uxx}},:,,, $5 $ ) ) 1B) t) V@rr_)r*rfenumrtypingrrr4tosa_serializerr+*executorch.backends.arm.tosa.specificationrrlfloat64double complex64cfloat complex128cdoubleuint8int64longr3r rUrHr]r_rrrrs  H* MM LL OO LL  MM KK KK JJ *Kt*KZ   J%%PS@S@r