Ei ^SSKJrJrJr SSKrSSKJr SSKJr SSK J r SSK J r J r Jr SSKJr /SQr\"S S 9S \ S \S \S\S\4 Sj5r\"S S 9"SS\55r\"S S 9SS \ S\\\R,SS4Sjj5r\"S S 9S\ S\4Sj5r\"S S 9S \ S\ S\4Sj5rg))Any NamedTupleOptionalN) compatibility)Graph) GraphModule)map_argNodeTarget) ShapeProp)replace_target_nodes_with size_bytesget_size_of_all_nodesget_tensor_metaget_size_of_nodeF)is_backward_compatible fx_moduleold_op old_targetnew_op new_targetc2^ [5n0m URRHnURU:XaURU:Xa[ UR U 4Sj5n[ URU 4Sj5n[U[5(d[S[U535e[U[5(d[S[U535eURX4XxUR5T U'MURUU 4Sj5T U'M XPlg)zModifies all nodes in fx_module.graph.nodes which match the specified op code and target, and updates them to match the new op code and targetc>TU$Nnval_maps _/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/fx/passes/graph_manipulation.py+replace_target_nodes_with..#s  c>TU$rrrs rr r!$s GAJr"zExpected tuple, got zExpected dict, got c>TU$rrrs rr r!-s  r"N)rgraphnodesoptargetr argskwargs isinstancetupleAssertionErrortypedict create_nodename node_copy) rrrrr new_graphnoder)r*rs @rr r sI "G%% 77f  !:499&:;DT[[*>?FdE**$';DJ<%HIIfd++$':4<.%IJJ%11D$))GDM&//6JKGDM& Or"c*\rSrSr%\\S'\\S'Srg)r1 output_size total_sizerN)__name__ __module__ __qualname____firstlineno__int__annotations____static_attributes__rr"rrr1s Or"rr)returncUb[U5R"U6 URRH%nURS:Xa g[ X5UlM' g)zGiven a fx graph module, update each node with its total size (weights + bias + output) and its output_size(output). For a non-module node, the total size is the output size. return total sizeNoutput)r propagater%r&r'rr)rr)r4s rrr7sV )&&-%% 77h   +9;& r"r4chURRS5nU(d[SUS35eU$)N tensor_metazNode zQ has no tensor metadata associated with it! Check that shape propagation has run.)metaget RuntimeError)r4rEs rrrIs=))-- .K D64 5  r"cSnURS:XaT[UR55nX1RnUR 5nUHupgX'R 5- nM [ U5nURR 5n X)- nUR(a.[R"/URS9R5n O-[R"/URS9R5n X-n X-n [X5$)zGiven a node with node.dtype and node.shape, return its total size and its output size. total_size = weights + bias + output_size r call_module)dtype)r'r/ named_modulesr(named_parametersnumelrshape is_quantizedtorch_empty_affine_quantizedrK element_sizetensorr) rr4total_num_of_elemssubmodule_dict submodule parameters_nameprE output_elemsize_per_elem_bytesr8r7s rrrVs  ww-i5578";;/ //1 "HE '') + #"$'K##))+K%#;; k'' ,. $ll2[5F5FGTTV$9J%3K k ..r"r)typingrrrrQtorch.fx._compatibilityrtorch.fx.graphrtorch.fx.graph_moduler torch.fx.noder r r torch.fx.passes.shape_propr __all__strr rlistTensorrrrrr"rrgs%,, 1 -//0 e,         - 6e,- e,AE  "*4 +=">   - "e, $ 3 - e,/ /4/J/-/r"