iixSSKrSSKrSSKJrJrJrJrJrJrJ r J r SSK r SSK J r SSKJr SSKJr SSKJr SSKJr SSKJr SS K Jr SS KJr SS KJr S S KJrJrJr S SK J!r!J"r"J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r* S SK+J,r, \"S5SS\ \RZ\\.\RZ4\\\.\4\\\.\4\.4S\\ \\/\ R`S4\\.\\/\ R`S444S\S\\.S\\ \\\.\44S\\ \\.\4S\\.S\1SS4Sjj5r2\"S5"SS55r3g)N)AnyCallableDictListOptionalSequenceTupleUnion)EdgeProgramManager) experimental)ExecutorchProgramManager)Program)save_pte_program)tabulate)nn)ExportedProgram) GraphModule) ExportRecipeLoweringRecipeQuantizationRecipe) EdgeProgramManagerTransformStageEdgeTransformAndLowerStageExecutorchStagePipelineArtifact QuantizeStageSourceTransformStageStageToBackendStage ToEdgeStageTorchExportStage) StageTypezfThis API and all of its related functionality such as ExportSession and ExportRecipe are experimental.modelexample_inputs. export_recipenamedynamic_shapesconstant_methods artifact_dirgenerate_etrecordreturn ExportSessionc F[UUUUUUUUS9nUR5 U$)a Create and configure an ExportSession with the given parameters. This function provides a convenient way to create an ExportSession and optionally run the export process in one step. Args: model: The PyTorch model(s) to export. Can be: - nn.Module or Dict[str, nn.Module]: Eager PyTorch model(s) - GraphModule or Dict[str, GraphModule]: Quantized model(s) (e.g., from prepare/convert) - ExportedProgram or Dict[str, ExportedProgram]: Already exported model(s) - str: Path to load an ExportedProgram from disk example_inputs: Example inputs for the model(s), either a list of input tuples or a dictionary mapping method names to lists of input tuples. First sample (index 0) is used for torch.export.export() to export the model. All samples are used as calibration dataset in PT2E Quantize stage. Optional when model is ExportedProgram (not needed). export_recipe: Contains the configuration for the export process name: Optional name for the export dynamic_shapes: Optional dynamic shape specifications constant_methods: Optional dictionary of constant methods artifact_dir: Optional directory to store artifacts generate_etrecord: Optional flag to generate an etrecord Returns: A configured ExportSession instance with the export process completed if requested )r#r$r%r&r'r(r)r*)r,export) r#r$r%r&r'r(r)r*sessions W/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/export/export.pyr.r.'s8j%# %)!+ G NN Nc<\rSrSrSrS)S\\R\\ \R4\ \\ \ 4\ \\ \ 4\ 4S\ \\ \\R S4\\ \ \\R S444S\S\ \ S \ \\\\ \44S \ \\\ \4S \ \ S \ \S S4SjjrS\\R\ \ \4S \ 4SjrS \ \4SjrS\ \S \\\44SjrS\S\S S4SjrS\S \ \4SjrS \\\44SjrS\ \S S4SjrS*SjrS*Sjr S \\\!44Sjr"S+S\ S \ 4Sjjr#S,Sjr$S \%4Sjr&S \'4S jr(S \)4S!jr*S"\ S S4S#jr+S+S\ S \,\R S44S$jjr-S-S\ S\ \,\R S4S \.\4S%jjr/S*S&jr0S \4S'jr1S(r2g).r,ka Manages the export process for ExecuTorch models. This class handles the export process through a pipeline of stages: 1. (Optional) Quantize - Apply post-training quantization to the model 2. Export - Export PyTorch model to ExportedProgram 3. EdgeTransformAndLower - Transform and lower to EdgeProgramManager 4. Executorch - Convert to ExecutorchProgramManager for final execution Nr#r$.r%r&r'r(r)r*r+c  [U[5(a8[RR U5n[ R "SU35 URU5Ul[U[5(aUOSU0Ul URS:XaCU=(d 0Ul [UR[5(aSUR0Ul O'B$+    3 3     / /     / / O43M3M3O  8<7I7I  ! !8  #22"22 0!00 (!2- GI r1cX[U[5(a3[[UR 555nUR U5$[U[ 5(ag[U[5(ag[U[R5(ag[S[U535e)z Detect the type of input model. Args: model: Input model in various formats Returns: String indicating the model type: "nn.Module", "GraphModule", or "ExportedProgram" rr nn.ModulezUnsupported model type: ) r7r?nextitervaluesr=rrrModule TypeErrortype)rQr# first_values r0r= ExportSession._detect_model_types eT " "tELLN34K**;7 7 e_ - -$ { + + ryy ) )6tE{mDE Er1cD/nURS:Xa/UR[R[R/5 URS:waUR [R 5 UR[R[R/5 U$)zz Get default pipeline stages based on input model type. Returns: List of stages appropriate for the input model type rUr) r>extendr"SOURCE_TRANSFORMQUANTIZEappend TORCH_EXPORTTO_EDGE_TRANSFORM_AND_LOWER TO_EXECUTORCH)rQstagess r0rK#ExportSession._get_default_pipelines  ! ![ 0 MM..&&   ! !%6 6 MM)00 1  55''   r1rfc20nSnU=(d UR5GHnU[R:Xa[UR5nGOU[R :Xa[ UR5nGOU[R:XaWSnURRb[URR5n[XPRRS9nGO!U[R:Xa![R"UR 5nOU[R":Xa![$R"UR 5nOU[R&:Xa![(R"UR 5nOU[R*:Xa![,R"UR 5nOMU[R.:Xa [1URR25nO[4R6"US35 U(dGMX2U'GM U$)z/Build the stage registry from the given stages.N)strictz> is unknown, you have to register it before executing export())rKr"r`rrGrarrcrEaten_transform_passesrBr!rirdr from_reciperITO_EDGEr EDGE_PROGRAM_MANAGER_TRANSFORMr TO_BACKENDrrerexecutorch_backend_configr;r<)rQrfstage_registrystage stage_typerjs r0rMExportSession._build_stages(s13 @D$>$>$@@JY777,T-?-?@y111%d&8&89y555(,%&&<<H,0++AA-)))2E2E2L2LyDDD2>>t?T?TUy000#//0E0EFyGGG8DD))y333&2243H3HIy666'(;(;(U(UV !l"`au-2z*AABr1rrrqc X RU'g)z Register a new stage or override an existing stage implementation. Args: stage_type: The type of stage to register stage: The stage instance to register NrN)rQrrrqs r0register_stageExportSession.register_stagePs,1Z(r1c8URRU5$)z Get a registered stage by its type. Args: stage_type: The type of stage to retrieve Returns: The registered stage instance, or None if not found )rNget)rQrrs r0get_registered_stage"ExportSession.get_registered_stageZs##'' 33r1cUR$)z\ Get all registered stages. Returns: Dictionary mapping stage types to stage instances rurQs r0get_all_registered_stages'ExportSession.get_all_registered_stagesfs###r1c U(d [S5eURS:Xa`[R[R1n[ U5U-nU(a*SR SU55n[SUSUS35eO~URS:Xan[R[R[R1n[ U5U-nU(a*SR S U55n[SUS US 35eUS nURRU5nUc[S US35eUR(d[S US35e[S[U55HlnXnXS- n URRU5n U c[S US35eU Rn U (dMPX;dMW[SU SUSUSU 35e g)NzPipeline stages cannot be emptyrz, c38# UHoRv M g7fNr&.0ss r0 .|'K8J18Jz Cannot run zK stage(s) with GraphModule input. GraphModule is already quantized. Remove z- from pipeline_stages or use nn.Module input.rc38# UHoRv M g7frrrs r0rrrrzR stage(s) with ExportedProgram input. ExportedProgram is already exported. Remove z9 from pipeline_stages or use nn.Module/GraphModule input.rStage zB not found in registry, register it using session.register_stage()z cannot start a pipeline. rzD not found in registry, , register it using session.register_stage()zInvalid transition from z to z. Valid predecessors for z: )rCr>r"r`rasetjoinrcrNrycan_start_pipelinerangelenvalid_predecessor_stages) rQrfincompatible_stagesfound_incompatible stage_names first_stagefirst_stage_instancei current_stageprevious_stagestage_instancevalid_predecessorss r0_validate_pipeline_sequence)ExportSession._validate_pipeline_sequenceos'>? ?  ! !] 2#,#=#=y?Q?Q"R !$V/B!B !"ii'K8J'KK  !+/)]*WY" # #'8 8**""&&#  "%V/B!B !"ii'K8J'KK  !+/)]*ceQi #3377 D  ' %gh $66vk]2LMN Nq#f+&A"IM#E]N"1155mDN% ]O+op"0!H!H "!n&N .~.>d=/R..;_B?Q>RT'r1cLURURS9 [URURS9nURHnUR R U5nUc[SUS35e[R"SU35 [R"5nURU5 [R"5U- S-nUR5nURS[U55 [R"SUS35 XR U'M g) N)rf)datacontextrz not found in registryzExecuting stage: i duration_msz execution done)rrLrr@rOrNryrCr;r<time perf_counterrun get_artifacts add_contextintrP)rQcurrent_artifactrrrqstartelapseds r0 _run_pipelineExportSession._run_pipelines (((( ) ,dFWFWX//J((,,Z8E} 6*5K!LMM LL,ZL9 :%%'E II& '((*U2d:G$224   ( (G E LL6*_= >3C $ $Z 00r1c$UR5 g)a Execute the full export process. This method orchestrates the export process with optional quantization: 1. (Optional) Apply quantization to the model 2. Export the PyTorch model to ExportedProgram 3. Transform and lower to EdgeProgramManager 4. Convert to ExecutorchProgramManager N)rr}s r0r.ExportSession.exports r1cUR$r)rPr}s r0get_stage_artifacts!ExportSession.get_stage_artifactss'''r1 method_namec URR[R5nUb URc [ S5eURnX;a([ SUS[UR5535eX1$)aa Get the ExportedProgram for a specific method after torch export. Args: method_name: Name of the method to get exported program for, defaults to "forward" Returns: The ExportedProgram for the specified method Raises: RuntimeError: If torch export stage has not been run KeyError: If the method name is not found in exported programs z@Exported program is not available. Run Torch Export Stage first. Method name 'z5' not found in exported programs. Available methods: ) rPryr"rcr RuntimeErrorKeyErrorrBkeys)rQrartifactexported_programss r0get_exported_program"ExportSession.get_exported_programs++// 0F0FG  x}}4R %MM  / }-&&*+<+A+A+C&D%EG  !--r1cJ[R[R[R[R4HWnUR R U5nUcM#URcM2[R"SU35 URs $ [S5e)a Get the EdgeProgramManager after edge lowering stages. This method checks multiple stages in order of preference: 1. TO_EDGE_TRANSFORM_AND_LOWER (combined stage) 2. TO_BACKEND (separate stage with backend delegation) 3. EDGE_PROGRAM_MANAGER_TRANSFORM (separate stage after TO_EDGE) 4. TO_EDGE (separate stage without backend delegation) Returns: The EdgeProgramManager Raises: RuntimeError: If no edge stage has been run z*Returning edge program manager from stage zEdge program manager is not available. Run one of the edge stages first: TO_EDGE_TRANSFORM_AND_LOWER, TO_EDGE, EDGE_PROGRAM_MANAGER_TRANSFORM, or TO_BACKEND.) r"rdrnrmrlrPryrr;r<r)rQrrrs r0get_edge_program_manager&ExportSession.get_edge_program_managers$  1 1   4 4     J //33J?H# (A I*VW}}$  E  r1c6UR5R$)z Get the ExecutorchProgram from the ExecutorchProgramManager. Returns: The ExecutorchProgram Raises: RuntimeError: If the executorch program manager is not initialized )get_executorch_program_managerexecutorch_programr}s r0get_executorch_program$ExportSession.get_executorch_programs224GGGr1cURR[R5nUb URc [ S5eUR$)z Get the ExecutorchProgramManager. Returns: The ExecutorchProgramManager Raises: RuntimeError: If the executorch program manager is not initialized zJExecutorch program manager is not initialized. Run Executorch Stage first.)rPryr"rerr)rQrs r0r,ExportSession.get_executorch_program_manager&sK++// 0G0GH  x}}4\ }}r1c6UR5R$)z Get the PTE buffer as bytes. Returns: The PTE buffer as bytes Raises: RuntimeError: If the executorch program manager is not initialized )rbufferr}s r0get_pte_bufferExportSession.get_pte_buffer7s224;;;r1 output_namecTU(dS5e[UR5U5 g)zc Save the model to a .pte file. Args: output_name (Optional[str]): The name of the .pte file. zNeed a valid output nameN)rr)rQrs r0 save_to_pteExportSession.save_to_pteCs%666{<<> Lr1cXR;a[SUS35eURUnU(d[SUS35eURUS$)aD Get the example input for a specific method. Args: method_name: Name of the method to get example input for, defaults to "forward" Returns: Tuple of tensors representing the example input Raises: KeyError: If the method name is not found in example inputs ValueError: If the example inputs list is empty rz' not found in example inputszExample inputs list for method z is emptyr)rArrC)rQrexample_inputs_lists r0get_example_inputExportSession.get_example_inputMsh 22 2];-7TUV V#22;?">{m9UV V##K033r1cPSSKJnJn UR 5nUR UR 5URS9nURU5nUc[SUS35eUcURU5nURU5$![an[S5UeSnAff=f)a. Run a specific method with the given inputs. Args: method_name: Name of the method to run, defaults to "forward" example_inputs: Optional inputs to use, defaults to the example inputs Returns: The outputs of the method execution Raises: RuntimeError: If the method cannot be loaded r)Runtime Verificationzexecutorch.runtime is not available. In OSS: Please ensure executorch is properly installed via pip. In fbcode: Please add //executorch/runtime:runtime to your dependencies.N) verificationzFailed to load method 'z' from the program) executorch.runtimerrModuleNotFoundErrorry load_programrMinimal load_methodrrexecute) rQrr$rre et_runtimeprogramr5s r0 run_methodExportSession.run_methodhs&  @[[] ))    ! 0D0D* %%k2 ?)+6HI   !!33K@N~..)# %[   sB B% B  B%c[[UR5[R[R 1-5nU(d [ S5 URRUS5nUc [ S5 URSS5nU(a>[UR55 UR5n[[USSS95 g[S 5 g) z8 Print delegation information for the exported program. zRNo delegation info available, atleast one of the lowering stages should be presentrNz:No delegation info available, run the lowering stage firstdelegation_infor fancy_grid)headerstablefmtzNo delegation info available)rBrrLr"rdrnrrPry get_contextprint get_summary!get_operator_delegation_dataframer)rQlowering_stagestage_artifactrdfs r0print_delegation_info#ExportSession.print_delegation_infos %% &44i6J6JK L  d 1155nQ6GH  ! U V)445FM  /--/ 0 BBDB (2v E F 0 1r1c>UR5R5$)z Get the etrecord from the ExecuTorchProgramManager. Returns: The etrecord in the ExecuTorchProgramManager Raises: RuntimeError: If the ExecuTorchManager is unavailable, or etrecord is not available in the ExecuTorchProgramManager )r get_etrecordr}s r0rExportSession.get_etrecords224AACCr1) rDrArEr>rIr@rLrGrOrNrPNNNNNNF)r+N)r5)r+r )r5N)3__name__ __module__ __qualname____firstlineno____doc__r rrYrr8rrrrtupler9TensorrrrboolrRr=r"rKrrMrvrzr~rrr.rrrrrrr rbytesrrr rrrrr__static_attributes__r1r0r,r,ks 4 &*"?CAE&*,1-rI II bii  k! "  o% &    rI! U5<<,-.S$uU\\3%67889;  rI"$#rI$sm%rI&!sDcN':!;<'rI(#5c8m)<#=>)rI*sm+rI,$D>-rI. /rIhF299k?DHIF F6tI@&DO&Y=M8N&P1151T1 4y 4Xe_ 4$4 50@+A$AYA AFD4 (T)5E*E%F(..O.: B H H0H" < <MsMtM"+44 u||S !4:%=A)/)/!u||S'8!9:)/ # )/V26 Dc Dr1r)4r;rtypingrrrrrrr r r9executorch.exirr executorch.exir._warningsr executorch.exir.programr executorch.exir.schemar&executorch.extension.export_util.utilsrrr torch.exportrtorch.fxrreciperrrrfrrrrrrrrr r!typesr"rYr8rrrr.r,rr1r0r s NNN .2<*C( DD   l$ "&;?=A"&#+>   S"))^ S+  S/ !"   > u||S() * d5s!2344 5 7 >  !>" 3-#>$U3S#X#678%>&uT#x-%89:'>(3-)>*+>,->>BlJ DJ DJ Dr1