i3> %SSKrSSKJr SSKJr SSKrSSKrSSKJ r J r J r SSKJrJr SSKJr SSKJrJr SSKJrJr SS KJr SS KJrJrJrJr SS K J!r! SS K"J#r# SS K$J%r%J&r& SSK'J(r( SSK)J*r* SSK+J,r,J-r- SSK.J/r/ \"5r0S=S\RbRdS\3\4S4S\5S\\6\RnRpS\(4 Sjjr S>S\RbRdS\3\4S4S\S\5S\RrRt4 Sjjr;S>S\(S\S\5S\RrRt4Sjjr<S>S\RrRtS\5S\RrRt4Sjjr=S \(S\S\(4S!jr>S?S\RbRdS\3\4S4S\S"\\6\3\4S4S\5S\RrRt4 S#jjr?S@S\RbRdS\3\4S4S\\S"\\6\3\4S4S\5S$\\6\@S\(4S%jjrA\RRRR\RRRR\RRRR\RRRR\RRRR\RRRR\RRRR\RRRR/rM\6\RnRp\NS&'\RRRR/rO\6\RnRp\NS''SAS(\(S\5S)\\P\@\44S*\\6\RnRpS\4 S+jjrQSAS\RbRdS\3\4S4S\5S)\\P\@\44S*\\6\RnRpS\4 S,jjrRSBS\RbRdS\3\4S4S\\S\5S)\\P\@\44S"\\6\3\4S4S*\\6\RnRpS\4S-jjrSSCS\(S\5S.\TS\4S/jjrUSCS\RbRdS\3\4S4S\5S.\TS\4 S0jjrVSCS\RbRdS\3\4S4S\5S.\TS\4 S1jjrWSDS\RbRdS\3\4S4S2\\@S.\TS3\TS4\5S5\5S6\\S\5S\4S7jjrXSDS8\(S2\\@S.\TS3\TS4\5S5\5S6\\S\5S\4S9jjrYS:\S;\S2\SS4S<jrZg)EN)Path)Optional)prepareQuantizedInputWrappertrace)CadenceMemoryPlanningprint_memory_planning_info) QuantFusion)CadenceDefaultQuantizerCadenceQuantizer)get_default_memory_config MemoryConfig)generate_etrecord)EdgeCompileConfigEdgeProgramManagerExecutorchBackendConfigExecutorchProgramManager) ToOutVarPass)ConstraintBasedSymShapeEvalPass) _transformto_edge)ExportedProgram) convert_pt2e)apply_exir_ops_passesapply_torch_ops_passes)print_ops_infomodelinputs. dump_graphs ops_to_keepreturncUc/n[XSSUS9nU(aN[R"S5 [R"URRR 55 U$)z< Trace the model with export and return an ExportedProgram. FT)is_qatstrictr!zGraph before quantization:)trace_fnlogginginfo graph_modulegraph print_tabular)rrr r!programs g/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/backends/cadence/aot/compiler.pyrr4sZ  eDkG 12 W))//==?@ N quantizercNUR5n[XX4S9n[XRUS9nU$)aO Trace and Prepare a model using the given quantizer. The quantizer must be supplied and be the same as the one used to fuse the model later, if applicable. If you do not expect that behavior, please use quantize_pt2 instead, which will instantiate a default quantizer for you if needed. Returns a GraphModule with the prepared model. r r!r )&get_ops_to_preserve_from_decompositionrprepare_traced_pt2)rrr/r r!traced_programprepared_programs r- prepare_pt2r7Js;BBDK ;N*{ r.r,c[XSS9nU(aD[R"S5 [R"URR 55 U$)aE Prepare a model using the given quantizer. The quantizer must be supplied and be the same as the one used to fuse the model later, if applicable. If you do not expect that behavior, please use quantize_pt2 instead, which will instantiate a default quantizer for you if needed. Returns a GraphModule with the prepared model. F)r$zGraph after preparation:) prepare_fnr'r(r*r+)r,r/r prepared_models r-r4r4dsA 5AN /0 ^))779: r.r)c[U5nU(aD[R"S5 [R"URR 55 U$)zC Convert the model Returns a GraphModule with the converted model. zGraph after convert:)rr'r(r*r+)r)r converted_models r- convert_pt2r={s?#<0O +, _**88:; r.converted_programcURVs/sHo"RPM nn[U[U55nU$s snf)a7 Fuse a converted exported program using the given quantizer. The quantizer must be the same as the one used to convert the model. If you do not expect that behavior, please use quantize_pt2 instead, which will instantiate a default quantizer for you if needed. Returns an ExportedProgram with the fused model. ) quantizerspatternrr )r>r/qpatterns fused_programs r-apply_pre_edge_transform_passesrEs?$-#7#78#7a #7H80+h2GHM 9s=calibration_datac>UR5 UR5n[XXES9nU(aD[R"S5 [R"UR R 55 [XbUS9nUcU/nUHnU"U6 M [XtS9n U $)Nr1zGraph after trace:r2) evalr3rr'r(r*r+r4r=) rrr/rFr r!r, prepared_gmsamples converted_gms r-get_fake_quant_modelrLs JJLBBDKE{TG )* W]]0023%W[QK"8#W${DL r.quant_input_argscZU(d [5n[UUUUUS9nUb [Xe5n[RR XaSS9n[ Xr5nU(aN[ R"S5 [ R"URRR55 U$)a Trace, prepare, convert and fuse the model using the given quantizer. If calibration data is provided, it will be used to calibrate the model. If not, the inputs will be used for calibration instead, which is useful for unit tests but should not be used for end-to-end use cases. Returns an ExportedProgram with the quantized model. Note: this function should not be called directly in general. Please use quantize_and_export_to_executorch for most needs. r/rFr Tr%z$Graph after quantization and fusion:) r rLrtorchexportrEr'r(r)r*r+) rrr/rFr rMrKr,rDs r- quantize_pt2rSs$ +- ( ) L#,\L ll!!,t!DG3GGM ;< ]//55CCEF r.TO_EDGE_OP_EXCEPTION_LISTTO_EDGE_PRESERVE_OPS expo_programconstant_methodscore_aten_exceptionsc 2[U5n[U[S[U=(d /-[S9US9nU(a\[ R "S5 [ R "UR5RRR55 U$)zA Lower an ExportedProgram to an EdgeProgramManager (in edge IR). T)_skip_dim_order_core_aten_ops_exception_list preserve_ops)compile_configrWzGraph after Edge lowering:) rrrrTrUr'r(exported_programr)r*r+)rVr rWrXedge_prog_managers r-_lower_ep_to_edger`s*,7L ( *C#)r++-  *  12  . . 0 = = C C Q Q S  r.c[U[RR5(dS5e[ X5n[ XRX45nU$)Nzmodel should be an nn.Module) isinstancerQnnModulerr`)rrr rWrXrVr_s r-export_to_edgere&sL eUXX__ - -M/MM -'L*#3 r.c4[UUUUUS9n[UUUUS9$)z; Trace, quantize and lower a model/inputs pair to edge IR. rO)r rWrX)rSr`)rrr/r rWrFrXquantized_models r-quantize_and_export_to_edgerh:s8# ) O )1  r. opt_levelc.[XS9n[X#5nU$)zV Lower an existing ExportedProgram to edge IR and apply frontend optimization passes. r2)r`r)r,r rir_cadence_prog_managers r-_lower_ep_to_cadencerlVs *'K0N r.c0[XUS9n[X45nU$)Nr2)rer)rrr rir_rks r-export_to_cadencerncs" 'u+N0N r.c.[X5n[UUUS9$)z_ Trace, quantize, lower a model/inputs pair to edge IR and apply frontend optimization passes. )rir )rSrl)rrr rirgs r-quantize_and_export_to_cadencerpns$#51O  r. output_dirmem_algoalloc_graph_inputalloc_graph_output memory_configc ^[RRXSS9n [U UUUUUUUS9$)NTrP)rqrirrrsrtrur )rQrR!_lower_ep_to_cadence_gen_etrecord) rrrqrirrrsrtrur eps r-!export_to_executorch_gen_etrecordrysB   U4  8B , +-#  r.rxc [X5n[X(5n [UR5RU R5R5 Uc [ 5n[ UUUUUS9n U R[U S[5S[5S95n [U UUUU5 U(a[X[U55 U $[R"S5 U $)N)rirrrsrtF)memory_planning_passemit_stacktraceto_out_var_passextract_delegate_segmentssym_shape_eval_passz:No output directory provided, skipping ETRecord generation)r`rrr^r)r r to_executorchrrrr _gen_etrecordrr'warning) rxrqrirrrsrtrur r_rkr{ exec_progs r-rwrws*":0N**,99--/<< 13 0+- +?*L*L!5!(N&+ ? A  +I 'D4DE  TU r. edge_program et_programcUS- n[UUUS9 [R"SU35 g![a [R"S5 gf=f)Nz etrecord.bin) et_recordedge_dialect_programexecutorch_programzGenerated ETRecord at z/Encountered exception while generating ETRecord)rr'r( Exception exception)rrrq etrec_paths r-rrs[ n,J M !-)  -j\:; MKLMs$, AA)FN)F)NF)NNFN)FNN)NFNNN)Fr)NrrTTNF)[r'pathlibrtypingr1executorch.backends.cadence.aot.ops_registrations executorchrQ.executorch.backends.cadence.aot.compiler_funcsrr9rrr&/executorch.backends.cadence.aot.memory_planningrr 5executorch.backends.cadence.aot.quantizer.fusion_passr 3executorch.backends.cadence.aot.quantizer.quantizerr r %executorch.backends.cadence.aot.utilsr rexecutorch.devtoolsrexecutorch.exirrrrrexecutorch.exir.passesr*executorch.exir.passes.sym_shape_eval_passr executorch.exir.program._programrrtorch.export.exported_programr'torchao.quantization.pt2e.quantize_pt2erpassesrrutilsrdefault_quantizerrcrdtupleobjectboollist_ops OpOverloadfx GraphModuler7r4r=rErLstrrSopsaten _linalg_detdefault _linalg_svd#_native_batch_norm_legit_functionallinearlinalg_vector_normunfoldanglerms_normrT__annotations__rUdictr`rerhintrlrnrpryrwrr.r-rsF8  N2 0V@9@A!+- 9=  88?? &#+ $uzz4456   4  88?? &#+     XX :  XX 2((&& XX.&2<@  88?? &#+  tE&#+$678     XX H-1;?,0 ) 88??) &#+ )())tE&#+$678 )  ) tCy) ))Z IINN&& IINN&& IINN66>> IINN!! IINN%%-- IINN!! IINN   IINN## :4 5 56  IINN##5d5::00148BF !tCK01#4 (=(=#>?   L48BF  88?? &#+ tCK01  #4 (=(=#>?   .-148;?BF 88?? &#+ ()  tCK01  tE&#+$678 #4 (=(=#>?<            88?? &#+         88?? &#+    ,!%"#,0 88?? &#+       L):!%"#,066 66 6  6  6L)666rM$M(MM Mr.