PhF ddlZddlmZddlZddlmZmZmZmZddl Z ddl Z ddl Z ddl Z ddl mcmZddlmZddlmZddlmZmZddlmZddlmZdd lmZmZdd lm Z m!Z!m"Z"d d lm#Z#gd Z$ejJGddZ&dZ'd!dZ(Gdde)Z*Gdde jVjXe*Z-de jVjXfdZ.de j^j@de j`jbfdZ2dZ3de jVjXde4dee jjdeee jlefeeeefffd Z7y)"N) defaultdict)DictListOptionalTuple) FakeTensor)SymInt)_PyTreeCodeGen _PyTreeInfo) ValueRanges)InputDim)ExportBackwardSignatureExportGraphSignature)ExportedProgramModuleCallEntryModuleCallSignature)!_check_input_constraints_pre_hook)rrrrrcVeZdZUeej ed<eej ed<y)CallSpecin_specout_specN)__name__ __module__ __qualname__rpytreeTreeSpec__annotations__iC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/_export/exported_program.pyrr1s foo &&v''r rc  d}i}|jjD]} | jdk(r||vr|jj| 5|jj ||} | j | | j } |jj| | | _| |||<ddd|dz }| jdk(sg} tj| jdD]} | j}||vr||jdd}||vsJ||}|jj| 5|jjtj j"j$j&|| fddd| j)| |jj| 5|jj+t-| }| j ||jj| ddd|jj/|j0t,k(rt3|j4dk(r|j4dj0t,k(rf|j4dj0t6k(rFt3|j4dj4t3|j4dj4z}nt3|j4}t9|Dcgc]}d| }}t;t=||||j_|jA|jjD]} | jd k(r| jBtj jDk(r`| j\}}}}tG||j}tG||j}i}g}tI|D]\}}|jB|jKvr|jB||<|jB|vr||jB}n4|jB|vr||jB}ntMd |jB|jO|jB||jO|jB||j)|||||f| _ tj|\} }tQ|||d|||tQ|||d|||| jd k(s| jBjRd k(s| j^}!}"}tG||!j}#i}$g}|jUD%cic]\}%}|%jdd|}&}%}tI|D]\}}|jB|jKvr|jB|$|<|jB|&vr|&|jB}n6|jB|vr||jB}ntMd |jB|#jO|jB||j)||!|"g|| _ tj|\} }tQ|#|$|d||||jj/|jjW|jA|S#1swYxYw#1swYxYw#1swYxYwcc}wcc}}%w) Nr placeholderroutput._arg_ call_functionzUnable to find value for map_impl),graphnodesopinserting_afterget_attrreplace_all_uses_withmeta erase_noder tree_flattenargsnamereplaceinserting_beforer)torchopsatencopy_defaultappendr$tuplelinttypelenchildren_specsdictranger r _codegen recompiletargetcondgetattr enumeratevalues RuntimeErrorregister_buffer_unliftritemseliminate_dead_code)'gminp_pos_to_param_buffer_namerr state_dicttensor_constantsbuffers_to_mutatecountbuffer_name_to_nodenode getattr_nodemetadatauser_output_nodes return_nodereturn_node_namebuffer_node_name buffer_node new_outputnum_argsinamespred true_graph false_graphoperandstrue_gmfalse_gm'inp_pos_to_param_buffer_name_for_submod real_operandsixoperandvaluer& body_graph num_mappedbody_gm!inp_pos_to_buffer_name_for_submodkeystate_dict_for_lookups' r!rNrN7s5 E 77m #44XX--d3#%88#4#44U;$L..|<#yyHHH''-(0L%O['(DU(KL4 QJE 77h  "  &22499=a@ #.#3#3 #'88'89I'J'R'RSVX['\$+/BBBB"56F"GK2248..!IINN0088; :T98 &,,[9 A**40XX__U3D-EF **:6##D) 10CRHHMMO   G " "#q(q!&&%/q!&&$. &&q)88 9 &&q)88 9 :  w--.!&x 1AtA3ZE 1&    BHHLLN 77o %$++*G6:ii 3D*k8b*//2Gr;#3#34H68 3M(2 G>>%A%H%H%JJBI..;B?~~3 *7>> : +;; 0 @*+FWW++GNNEB,,W^^UC!((1 3z; FDI,,];JAw 7 !  7 !  77o %$++*>*>**L04 -J Xb*//2G02 -M#-"2"2"4%"4JC C%u,"4 "% )2 G>>%A%H%H%JJ> 5gnn E +;; 0 @*-Fw~~FV+WXX++GNNEB!((1 3$Z@-@DI,,];JAw 1 ! IZHHMMOHH  "LLN Ii43098 102 2z%s3 A&[A [A[" [/*[4[ [ "[, cXi}|jD]\}}||jvrNd|vr8|j|jdd||jdd||<n|j||||jvsqd|vr8|j |jdd||jdd||<|j |||Zt |dkDrLt|dsJ|jD]+\}}||jvs|j|||||<-d}i}|jjD]} | jdk(s| j|jvr+|j| j} | |vr || ||<n| ||<| j|jvr+|j| j} | |vr || ||<n| ||<t|dr4| j|jvr|j| j||<|dz }|S)Nr%r&rlifted_tensor_constantsr#!inputs_to_lifted_tensor_constantsr)rObuffersrMr6 parametersregister_parameterrAhasattrrvr+r,r-r5inputs_to_buffersinputs_to_parametersrw) new_gmgraph_signaturerSrT#param_buffer_name_to_corrected_namer5rnrVrRrX buffer_name param_names r!'_construct_inp_pos_to_param_buffer_namersY+-'!'') e ?** *d{&&t||C'=uE<@LLc>>&&tU3<@3D94 E#%  "" 77m #yyO===-?? J "EEB!r r)rrrrrrs@r!rrs ??r r) metaclassplain_graph_modulecztj||j||}|jtd|S)NrT) with_kwargs)rrr+register_forward_pre_hookr)rrr stateful_gms r!_create_stateful_graph_moduler&sF&..  +1 /K ))*KY])^ r epreturnc tj|j}t||j|j |j }t|||jj|jj|j |j |jj}t||j|j}|jj!|jj|Sr)copydeepcopy graph_modulerrrSrTrN call_specrrrUrrrr1update)rr~rR unlift_gms r!%unlift_exported_program_lifted_statesr1s ]]2?? +F#J""BMM23F3F$ $     ,,F.fb6J6JBLcLcdI NN"//../ r c tjj||}|S#t$r\t j dtjj|tjj }||_Y|SwxYw)NzUnable to execute the generated python source code from the graph. The graph module will no longer be directly callable, but you can still run the ExportedProgram, and if needed, you can run the graph module eagerly using torch.fx.Interpreter.)r8fx GraphModule SyntaxErrorwarningswarnGraph_graph)rr+rQs r!_create_graph_module_for_exportrDsx XX ! !$ . I     G XX ! !$(8 9 I s $A!B B rnum_lifted_params_buffersexample_inputsc |jjdg}|jjdg}tt}i}t |j j D]Z\}}|jdk7rnF||k\s|||z } |t| j|j|||j<\g} tt} |D]} || dD]y}t|| d} | | jt| d| d| jdd x}sK||dD]$}t||d}| j| |f&{i}|Dcic]}||| }}| jD]\}}|Dcgc]}|j}}|Dcgc]}|j}}t!|}t#|}||ksJ||j$jd }t'|t(sJ|j*|j,}t'|t.sJd |d t1||j2j4}t||||<|| fScc}wcc}wcc}w) a- Process the constraints stored in the graph module to return something more readable. Args: graph_module (torch.fx.GraphModule): GraphModule returned from dynamo.export, which contains the "input_shape_constraints" and "inline_constraints" metadata example_inputs: Flattened list of example inputs used to export the graph module Returns: range_constraints (Dict[sympy.Symbol, ValueRanges]): Mapping of symbols (from SymInts) appearing in the fake tensors in node.meta["val"] to their range constraints, which are a tuple containing (lower, upper) constraints. equality_constraints (List[Tuple[InputDim, InputDim]]): List of tuples of (node, dim) to mark that these dimensions are equal. input_shape_constraintsinline_constraintsr#t_iddimminmaxsharedNvalzExpected SymInt but got z: )r1getrlistrJr+r,r-idr=r5r r rOlowerupperrr input_name isinstancershaperr r@rX_expr)rrrrrtensor_id_to_nodesplaceholder_nodesrbrX example_inputrmulti_range_constraints constraintnode_dimr other_nodeother_node_dimrsymbol input_dimmulti_range_constraintrcmin_valsmax_valsmin_valmax_valrsymints r!_process_constraintsrYs0+//334MrR%**../CRH0;4/@24\//5564 77m #  ) )*1/H+HIM r-0 1 8 8 C+/ dii (7=?BMTAR- &z&'9:Dj&78H $H - 4 4Ju-z%/@A  $$77v7"4VF^"DJ%-j&-%HN(//>0JK#E;. :<K]]J\!3F!;;J\].E-J-J-L) )(>>'=BHH'=>'=>'=BHH'=>h-h-'!!! 4 45::5A#z***9==)&&)^-EfXRPTU[P\~+^^)""$/$A&!.M" 2 22)^ ?>s( I4I9*I>r)8r collectionsr dataclassestypingrrrrrsympyr8torch.fxtorch.utils._pytreeutils_pytreertorch._subclasses.fake_tensorr%torch.fx.experimental.symbolic_shapesr torch.fx.graphr r torch.utils._sympy.value_rangesr @torch._export.passes.add_runtime_assertions_for_constraints_passr torch.export.graph_signaturerrtorch.export.exported_programrrrr__all__ dataclassrrNrr@rrrrrexportnnModulerrintTensorSymbolrrr r!rsK #.. $$4867  5  ((( [z5(p $ &?588//;V?ehh6J6Jell.J.Juxx&*U3((&&U3"U3&U3 4 k) *Dx7I1J,K KL U3r