IЦi+SSKrSSKrSSKJr SSKJrJrJrJrJ r J r J r J r SSK r SSKr SSKJr SSKJr /SQr\ \ \ R(\\ R(4r\ \ R(\4r\\ R.R0r\ \ R.R0r\\r1Skr\"SS 9S 5r\"SS 9S \\\ R>R@4S \ R.R0S \4Sj5r!\"SS 9S \ R.R0S \"4Sj5r#\"SS 9"SS55r$\"SS 9S\ R.RJS \ R.RJ4Sj5r&g)N) dataclass)AnyDictListMappingOptionalSetTupleUnion) compatibility)_get_qualified_name)get_acc_ops_nameget_node_targetis_node_output_tensorFxNetAccFusionsFinderlegalize_graph> call_method call_module call_functionF)is_backward_compatiblec[U[5(aU$UR(aSUR;aSUR3$URR SS5nU(aUOSSUR3$)Nacc_opsacc_ops.z torch._opsz torch.ops.) isinstancestr __module____name__replace)kmodules [/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/fx/passes/tools_common.pyrrsl!S )q||3!**&&%% + #&+1QZZL99 submodulesnodereturncDUR[;d-SSR[5-SUR3-5eURS:XaP[UR[ 5(deXRn[ US[U55n[U5$URS:XaCURnURbSUR;aSUR3$[U5$[UR[ 5(deUR$) a  Given a `node` returns its target typename. For "call_method" node, return node.target which is the name of that method being called. This could potential lead to conflict but should be okay because normally it's on a tensor. For "call_function" node, return typename of node.target. For "call_module" node, return typename of the module that node.target point to. If seeing "_VariableFunctionsClass" in the target name string, it will be replaced by "torch". e.g. _VariableFunctionsClass.relu would become torch.relu. zExpect op types of z, z , but found r_base_class_originrrr) opCALLABLE_NODE_OPSjoinrtargetrgetattrtyperrrr )r%r&submod submod_typer-s r#rr*s$ 77' ' *; <<dggY?WW ' ww-$++s++++KK(f&:DLI  ,, O #kk  ,f>O>O1Ov' ( %V, $++s++++{{r$cURRSS5nUSL=(a [U[R5$)a Checks if the node output produces a Tensor or not. NOTE: This requires to run `ShapeProp` on the containing fx graph before calling this function. This is because it works by checking the `type` metadata on the node. This metadata is produced by the `ShapeProp`. r/N)metaget issubclasstorchTensor)r&type_s r#rrQs3 IIMM&$ 'E   @E5< C vv..zz$|'@'@@(((&&|5H5H'RR%%c*14r$r'c@0n[UR5nUGHnX1;aM UR[;aM!SUR;aM3X0R;aMDUR UR RU5U1[UR5U1S9nUR(GaURR5nURUUR[5S9 SUR;anURH^nUR[;aMXTR ;aM*URU5 URUUR[5S9 M` URHnUR[;aMSUR;aM+XdR ;aM<URU5 [!UR"UR RU55UlURUUR[5S9 M UR(aGM[UR 5UR::d"U=RUR -slGM^UR HnUR X'M GM U$)N tensor_meta)rGr@rHrI)r[)r>r;r*r+r3rYr@r]setrNrIpopr^rHusersrPminrG)rAresultr;r&rZuserr_rOs r#__call__FxNetAccFusionsFinder.__call__sE/1( D~ww// )>>)>B>N>N!ZZ--d3f4//0$(6 ?O?L 111#66::<''  ''E(! 1 $ 77*;;$#5#55$$--d3//((//$'E0!+ //Cvv%66 $0 000  ))#.03$114::3C3CC3H1L-++$$++ #,01111T **+t~~=,"4"44%++A , 2 2FI,{@ r$)r;r"r@r=)rrrSrT__doc__r6fx GraphModulerWrBrrYr NodeListrr^rNoderirXrRr$r#rr]s #uxx33## F&* $9$gx'($'" $LD$uxx}}g56Dr$rgmcR^[R[R[R[R[R [R [R[R[R[R[R[R[RRR R"[RRR$R"[RRR&R([RRR*R"[RRR,R"/n[.R1UR2R4S5n[R6R95nUR2R4H"nUR:HnX%==S- ss'M M$ [<R>"5nUR2R4HnX$S:XdM URAU5 M 0m[CU5S:aURE5nURGUU4Sj5TU'UR:H]nX%==S-ss'X%S:XdMURHS:Xa#URJU;aURMU5 MLURAU5 M_ [CU5S:aM[CUR45[CUR2R45:a)[OSUVs/sHoBUS:wdM UPM sn35eUR2RPUl(X0lU$s snf)a Replace the graph of the given GraphModule with one that contains the same nodes as the original, but in topologically sorted order. This is used by the merge_matmul transformation below, which disturbs the topologically sorted order of its input GraphModule, so that this order is restored before further transformation. Arguments: gm: The graph module to topologically sort. It is modified in-place. Returns: The graph module in-place sorted rc>TU$r=rR)xenvs r# legalize_graph...s c!fr$rz&Input graph has cycles, unable to add ))operatorrKmulsubfloordivtruedivmodleltgegteqner6opsatensym_constrain_rangedefaultsym_constrain_range_for_size _assert_asyncmsg scalar_tensor_assert_scalardictfromkeysr?r@rlGraphre collectionsdequeappendlenpopleft node_copyr*r- appendleft RuntimeError_codegen) rpPRIORITIZED_OPSindeg new_graphr&rhqueuecurrus @r#rrst*            **22 33;; $$(( $$,, %%--#O( MM"((..! ,E IJJD K1 K +002E ;!  LL /1C e*q.mmo&&s,<=CIID K1 K{a77o-$++2P$$T*LL&  e*q. 9??c"((..114u5autVZP[_`P`du5a4b c  **IH I 6bs " N$ 2N$ )'rrx dataclassesrtypingrrrrrr r r r6torch.fxtorch.fx._compatibilityr torch.fx.noder __all__r7TensorsTensorOrTensorsrlrornrWrNamesr+rnnModulerboolrrrmrrRr$r#rs!HHH 1-  ell#T%,,%77 8 g-.   ehhmm  S Ce, :- :e,#UXX__,-#5:XX]]##-#Le,A A$A-Ae,UU-Upe,Guxx++G0D0DG-Gr$