i.SSKrSSKrSSKrSSKJrJrJr SSKJrJ r SSK J r SSK J r SSKJr SSKJrJr SSKJrJrJr SS KJr SS KJr "S S 5r"SS5rS\\-\ -S\4SjrS$S\\-\ -S\ S\ 4Sjjr!Sr"S\S\#4Sjr$SSSSS4S\S\#S-S\ S\ S\\/\#S-4S \\/\#S-44 S!jjr%S$S"\S\\-\ -S\ S\ 4S#jjr&g!\a \"S 5 ef=f)%N)AnyCallableType)EdgeProgramManagerExecutorchProgramManager)%_update_exported_program_graph_module)Verifier)ExportedProgram) GraphModuleNode)configconstsvisualize_from_config)ModelExplorerConfig)!PytorchExportedProgramAdapterImplzXError: 'model_explorer' is not installed. Install using devtools/install_requirements.shc|\rSrSr%SrSrS\R-\S'Sr \ \S'Sr SS\ S \ S-4S jjr S rS rS rg)SingletonModelExplorerServer"zSingleton context manager for starting a model-explorer server. If multiple ModelExplorerServer contexts are nested, a single server is still used. Nserverrnum_openg@open_in_browserportc[RciS/nU(dURS5 Ub+URS5 UR[U55 [R "U5[lggNzmodel-explorerz--no_open_in_browserz--port)rrappendstr subprocessPopenselfrrcommands t/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/devtools/visualization/visualization_utils.py__init__%SingletonModelExplorerServer.__init__,sa ' . . 6'(G"56x(s4y)2<2B2B72K ( / 7c[RS-[l[R"[R5 U$)N)rrtimesleepwait_after_startr s r" __enter__&SingletonModelExplorerServer.__enter__6s2 ( 1 1A 5 %- /@@A r%c[RS- [l[RS:Xaj[RbX[RR5 [RR [R 5 S[lggg![ Ra! [RR5 NBf=f)Nr'r) rrrkillwaitr*rTimeoutExpired terminater typevalue tracebacks r"__exit__%SingletonModelExplorerServer.__exit__=s ( 1 1A 5 %- ( 0 0A 5+22>,3388:D077<<4EE 7;,3? 6"00D077AACDs!-B2CCTN)__name__ __module__ __qualname____firstlineno____doc__rrr__annotations__rintr*boolr#r,r7__static_attributes__r9r%r"rr"sR '+FD:## #*HcLL3:L ;r%rcD\rSrSrSrSrS S\S\S-4SjjrSr S r S r g) ModelExplorerServerMz5Context manager for starting a model-explorer server.g@NrrcS/nU(dURS5 Ub+URS5 UR[U55 [R"U5Ulgr)rrrrrrs r"r#ModelExplorerServer.__init__RsN#$ NN1 2   NN8 $ NN3t9 % &&w/ r%cD[R"UR5 gN)r(r)r*r+s r"r,ModelExplorerServer.__enter__[s 4(()r%cURR5 URRUR5 g![R a URR 5 gf=frJ)rr/r0r*rr1r2r3s r"r7ModelExplorerServer.__exit__^sT  $ KK  T22 3(( $ KK ! ! # $s%A.A32A3)rr:) r;r<r=r>r?r*rBrAr#r,r7rCr9r%r"rErEMs,?003:0*$r%rE visualizablereturnc[U[5(aU$[U[[45(aUR 5$[ SU35e)Nz Cannot get ExportedProgram from ) isinstancer rrexported_program RuntimeError)rNs r"_get_exported_programrTfsM,00,!35M NOO,,.. 9,H IIr%F reuse_serverno_open_in_browserc [5n[RnURS[ U5US9 U(aUR 5 [ SURSU5US.UD6 g)a/Wraps the visualize_from_config call from model_explorer. For convenience, figures out how to find the exported_program from EdgeProgramManager and ExecutorchProgramManager for you. See https://github.com/google-ai-edge/model-explorer/wiki/4.-API-Guide#visualize-pytorch-models for full documentation. Executorch)rRsettingsr )r rVNr9)r rDEFAULT_SETTINGSadd_model_from_pytorchrTset_reuse_servervisualize_model_explorerpop)rNrUrVkwargs cur_configrYs r" visualizerapspJ&&H%%.|<& ##%zz(J/- r%c [S0UD6 g)z9Wraps the visualize_from_config call from model_explorer.Nr9)r)r_s r"r]r]s r%r` file_namec v[R"UR5S5n[R"US5SnUHnSUS'M SU[U5VVs/sH upTXES.PM snnS.n[ US5n[R "Xg5 S S S 5 g s snnf!,(df  g =f) zSave the graphs stored in the `cur_config` in JSON format, which can be loaded by the Model Explorer GUI. :param cur_config: ModelExplorerConfig containing the graph for visualization. :param file_name: Name of the JSON file for storage. graphs_listrgraphsrXcollectionLabel)graphlevel)labelrfgraphsWithLevelwN)jsonloadsget_transferrable_data enumerateopendump)r`rcrerfrhri json_datafs r"_save_model_as_jsonrus**Z>>@OPK ZZ A ' 1F#/  AJ6AR ARe ,AR I i  )     s"B$ B** B8c:URRSS5$)Ndelegation_tagmetagetnodes r"r}s$Jr%c:URRSS5$)Nclusterrxr{s r"r}r}sTYY]]9d3r%rRjson_file_nameget_node_partition_nameget_node_qdq_cluster_namec 8[5n[U[R5nUR 5n [ UR R5n U SSRn /n [XSS9HupSnU"U5=nbFUU ;aU RU5 U RU5nURSS5nUS US US 3- nU"U5=nb/[U5S:waUS- nURSS5nUS US 3- nXl M [UR5nURRU 5 S nSSUSU30nURRU5 Ub [!Xq5 gU(aUR#5 [%U40UDSU0D6 g)aVisualize exported programs using the Model Explorer. The QDQ clusters and individual partitions are highlighted. To install the Model Explorer, run `devtools/install_requirements.sh`. To display a stored json file, first launch the Model Explorer server by running `model-explorer`, and then use the GUI to open the json. NOTE: FireFox seems to have issues rendering the graphs. Other browsers seem to work well. :param exported_program: Program to visualize. :param json_file_name: If not None, a JSON of the visualization will be stored in the provided file. The JSON can then be opened in the Model Explorer GUI later. If None, a Model Explorer instance will be launched with the model visualization. :param no_open_in_browser: If `True`, a browser instance with the model explorer will NOT be launched, and only the URI to the model explorer server with the visualization will be printed. :param reuse_server: If True, an existing instance of the Model Explorer server will be used (if one exists). Otherwise, a new instance at a separate port will start. :param get_node_partition_name: Function which takes a `Node` and returns a string with the name of the partition the `Node` belongs to, or `None` if it has no partition. :param get_node_qdq_cluster_name: Function which takes a `Node` and returns a string with the name of the QDQ cluster the `Node` belongs to, or `None` if it has no cluster. :param kwargs: Additional kwargs for the `visualize_from_config()` function. rfrT)strictN/:z partition z ()z cluster (rXurlz graphs://rV)r rrrZconvertlistrhnodesziprindexreplacelen namespacere model_sourcesrur\r)rRrrVrUrrr_r`adapterrfrexplorer_nodesknown_partition_names explorer_noder|rpartition_name partition_idsafe_partition_name cluster_namesafe_cluster_name graphs_indexname model_sources r"visualize_with_clustersrsHJ0&11G__ F !''-- .EH%a(..N">F  5d; ;N H%::%,,^<066~FL"0"8"8S#  :l^26I5J!L LI5d; ;L H9~"S !- 4 4S!  9%6$7q9 9I"+? GDz--.L!!&) D(-4&,/P'QL##L1!J7   ' ' )    2 r% graph_modulec h"SS[5n[U5n[XU/S9n[XU40UD6 g)a#Overrides the graph_module of the supplied exported_program with 'graph_module' before visualizing. Also disables validating operators to allow visualizing graphs containing custom ops. A typical example is after running passes, which returns a graph_module rather than an ExportedProgram. c6\rSrSrSrS\\\S44SjrSr g) visualize_graph.._any_opi'ANY_OPrO.c[4$rJ)rr+s r"allowed_op_types1visualize_graph.._any_op.allowed_op_types*s ; r%r9N) r;r<r=r>dialecttuplerrrrCr9r%r"_any_opr's  eDIsN&; r%r)override_verifiersN)r rTrra)rrRrUrVr_rs r"visualize_graphrsA( --=><G9.@KFKr%)TF)'rmrr(typingrrrexecutorch.exirrr executorch.exir.program._programrtorch._export.verifierr torch.export.exported_programr torch.fxr r model_explorerr rrmodel_explorer.configr3model_explorer.pytorch_exported_program_adater_implr ImportErrorprintrrErTrBrar]rrurrr9r%r"rs  &&H,9&  DD9(;(;V$$2J!$669QQJJ$!$669QQ: $7 C <"&$= 4e %e $Je e  e &tfcDj&89 e  ( d  e V$ LL%(::=UULL L  b  s CC+