[Th<0 SSKrSSKJrJrJr SSKrSSKJr SSK J r /SQr "SS\RR5rS\RRS \4S jrS \RRS \S \4SjrSS\\R$R&\RR4S\\\RR(/\4S\S \4Sjjrg)N)CallableOptionalUnion)map_arg) split_module)FoldedGraphModuleget_unique_attr_name_in_modulesplit_const_subgraphsc ^\rSrSrSrS S\R RS\RRS\ \RRS\ \ S\ 4 U4Sjjjr U4S jr S rS rU=r$) ras FoldedGraphModule is a GraphModule which also contains another `const_subgraph_module` representing a subgraph which has all const attr inputs and which can be run once before running the main standard `graph`. The `const_output_names` are the ordered list names of attrs which represent what each respective output from the const_subgraph should be set on which attrs. rootgraphconst_subgraphfx_const_folded_attrs_namedevice_for_folded_attrsc>[TU]X5 UcSO[RR X5UlSUlX@lXPlg)NF) super__init__torchfx GraphModuleconst_subgraph_modulehas_folding_been_runrr)selfr rrrr __class__s X/var/www/auris/envauris/lib/python3.13/site-packages/torch/fx/experimental/const_fold.pyrFoldedGraphModule.__init__sO %% %%d; " %*!*D''>$c`>UR(dUR5 [TU] "U6$N)r run_foldingr__call__)rargskwargsrs rr"FoldedGraphModule.__call__-s(((    w&&rcx^TRb TRcgTR(aeSTlTR5nU4Sjn[U[5(a5[ R RUVs/sH o2"U5PM sn5OU"U5n[TTRU5 gs snf)NTcX>[RR[U[5(dUR 5R 5O-[R"U/5RTRS9[U[R5(aURS9$SS9$)N)deviceF) requires_grad) rnn Parameter isinstanceintdetachcloneTensortorr))irs r _create_param4FoldedGraphModule.run_folding.._create_paramCs88%%!!S))   "\\1#&))1M1M)N1;Au||1L1Laoo & SX & r) rrrr,tuplerr* ParameterListsetattr)r folded_attrsr3r2paramss` rr!FoldedGraphModule.run_folding2s  & & ...6 ,,,,$(! 113  ,.. HH " "l#KlM!$4l#K L|,  d55v> $Ls?B7)rrrr)NNcuda)__name__ __module__ __qualname____firstlineno____doc__rr*ModulerGraphrstrrr"r!__static_attributes__ __classcell__)rs@rrrs}4848'- ?hhoo?xx~~?!0 ? %-SM ? "% ??$' ??rrgminline_mod_namecv^ [UR55Un[U[RR 5(deSnUR RH)nURS:XdMURU:XdM'Un O UceURnURn0m SnU 4SjnUR RHn U RS:Xa,U RU;aXiROXWT U 'US- nM?U RS:Xa-U RSn [X5n URU 5 M|UR RU5 UR R!X5n SSS5 W T U 'M UR R#5 g!,(df  N2=f)z Given `gm` and some graph module which is called with target name `inline_mod_name`, this helper will inline all of the nodes from that called graph module into `gm`. N call_modulercP>TUnURR5UlU$r )metacopy)nodenew_nodereplacement_mappings rreplacement_fn&_inline_module..replacement_fnjs$&t, ( r placeholderoutput)dict named_modulesr,rrrrnodesoptargetr#r$namerreplace_all_uses_withinserting_before node_copyeliminate_dead_code)rFrG inline_modcall_mod_node_to_replacerM call_mod_argscall_mod_kwargsph_countrP inline_nodeoutputsoutput_replacementsrNrOs @r_inline_modulergSs b&&()/:J j%(("6"6 7 77 7# 77m # (F'+ $  $ // /-11M.55O>@H "''-- >>] *##6 0 01", , MH  >>X %!&&q)G")'"B  $ : :;N O  XX & &'? @xx))+FHA+3K('.*HH  " A @s "F** F8 mod_tracedrZreturncB[R"SSU5nUSR5(aSU3n[X5(aY[R"SU5nUcUS-nO'UR SS5up4US[ U5S-3n[X5(aMYU$)zH Make sure the name is unique (in a module) and can represents an attr. z[^0-9a-zA-Z_]+_rz (.*)_(\d+)$_1rS)resubisdigithasattrmatchgroupr-)rhrZrrbasenums rr r s 66"C .D Aw4&z * # #. =$;D Aq)IDV1SX\N+D * # # Krmoduleskip_folding_node_fnrc  ^^[U[RR5(d [RR U5nOUn[ 5mSnUR RHmTRS;aMTRS:wa+[ TR5RT5(dMPU(aU"T5(aMfTR5(aM}TRT5 TRS:wdMSnM U(d[X3R 5$S[RR4U4Sjjn[X0U5nSupxUR [#XhS5pU (aU R RO/H@mTRS :XdM[%UTR&[#U TR&55 MB U R RH@mTRS :XdM[%UTR&[#U TR&55 MB Sn UR RH3mTRS :XdMTR&U:XdM'TR(n O U ce[RRXiR 5n U R RGHmTRS :Xa[TR(S [*5n M3TRS :waME[-U4S jU 55nURS:XdeU R R/T5 U R R1UR&5nSSS5 TR2R55WlTR7U5 U R R9T5 GM S[;5;de[=US5n[%UUW (a[R>RA5O[R>RC55 UR RHmTRS :XdMTR&U:XdM'TR R/T5 TR R1U5nSSS5 TR2R55WlTR7U5 O [EXh5(a [GXh5 UR RI5 [UUR U R UU5$!,(df  GN=f!,(df  N=f)a2 Looks through `module` for any nodes that have all constant attribute inputs and separates them out into their own constant subgraph, and returns a FoldedGraphModule which runs that constant subgraph on the first run to set attributes on the module prior to running the non-constant portion of the graph. F>rTrRget_attrTrMc>UT;aS$S$)NrrS)rM const_nodess r mod_partition,split_const_subgraphs..mod_partitionsK'q.Q.r)submod_0submod_1NrIrTrrRc3^># UH"oRTR:XdMUv M$ g7fr )rZrY).0nrMs r (split_const_subgraphs..s"N"4Q$++8Mqq"4s- -multiple_outputs_FX_CONST_FOLDED_ATTRS)%r,rrrsymbolic_tracesetrrWrXall_input_nodesissubset is_impureaddrNoderrgetattrr7rYr#r5nextr\ryrKrLr[ erase_nodelocalsr r*r6r+rqrgr^)rvrwrrhfound_const_foldingr}splitconst_mod_namenon_const_mod_nameconst_gm non_const_gmcall_const_gm_args root_const_gmrin_noderNrr8r|rMs @@rr r s) fehh22 3 3XX,,V4  '*eK  && 77/ /  77j T-A-A)B)K)K * *   $8$>$>  >>     77j "& 1'6  -=-=>>/EHHMM/ ] ;E)?&N"^^WUPT-Ul -9 ""((b@ 77m # E4;; dkk(J KA$$ 77m # E4;;$++(F G% !! 77m #{{n,%)YY" "  )) )HH((?M##)) 77h )$))A,>   77m # N"4NNzzZ'''  1 1$ 7$**33GNNCH8 (  ""8,&&t,*  )) ) "@,"  "$4 %((:L:L:N  !! 77m # ~(E,,T2#zz223MN 3 $  0L   & &| 4  "u))u1 KK##%   "  G8 7*32s&UU U  U" )Ncpu)rntypingrrrtorch.fxr torch.fx.nodertorch.fx.passes.split_moduler__all__rrrrCrgr r*rArboolr r{rrrs ,,!5 ??,,??D1#uxx++1#c1#huxx/C/C3SV,GK#(U %((//588#7#77 8U"8UXX]]OT,A#BCU!U Ur