o "j@sddlmZddlmZddlZddlmZddlmZm Z m Z ddl m Z m Z er8dd l mZmZdd lmZd!ddZddZddZddZGdddZd"ddZdd ZdS)#) annotations) TYPE_CHECKINGN) to_sequence) InnerErrormap_if map_if_extend)SIRRuntimeCacheSymbol) Statement StatementIR)SymbolicTraceContextvalueslist[Symbol] | list[object]statedict[str, Symbol]cs$t|ddfddddd}|S)aH Replaces Symbol objects with their corresponding values. Args: values: A list of values that may contain Symbol objects. state: A dict mapping Symbol names to their corresponding values. Returns: A new list with Symbol objects replaced by their corresponding values in the state dict. cSs t|tSN isinstancer xrd/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/jit/sot/symbolic/interpreter.py. z replace_symbol..cs |jSrnamerrrrr/rcSs|Srrrrrrr0predZtrue_fnZfalse_fn)r)rrrrrreplace_symbols r"cCsJtjjrdSddlm}|j}|}t||D]}| ||qdS)Nr)core) paddlebaseZ frameworkZ use_pir_apiZpaddle.frameworkr#Zop_proto_and_checker_makerZkOpCreationCallstackAttrNamefor_each_ops_betweenZ _set_attr)startendstackr#Zop_makerZcallstack_attr_nameoprrr_append_opstack_between5s  r+ccs,tj}|j||}|EdHdSr)r$staticdefault_main_program global_blockops)r'r(programr/rrrr&Bs r&cCstj}t|jSr)r$r,r-lenr.r/)r0rrropnum_in_programJs r2c@sPeZdZdZdddZdd d ZdddZdddZddZddZ ddZ dS) Interpreterz; Interpreter is used to interpret and execute SIR. symbolic_contextrcCs ||_dSr)_context)selfr4rrr__init__Us zInterpreter.__init__rstrreturnr cCs |j|S)z Returns the StatementIR object by given name. Args: name: The name of the StatementIR. Returns: The StatementIR object with the given name. )r5get_sir)r6rrrrr:Xs zInterpreter.get_sirrrcs||}|jD]G}t}t|j}t||j||}fddtt|tt|j kr3t dt |td|j t ||j ddfddddd qt|j S) aD Runs the StatementIR with the given name using the provided state. Args: name: The name of the given StatementIR to run. state: A dict mapping Symbol names to their corresponding values. Returns: A list of the Symbol of the StatementIR after execution. cs||j<dSrrvsrrr_setvsz!Interpreter.run_sir.._setz*Number output mismatch, some error happen.r cSs t|tSrrr;rrrrrz%Interpreter.run_sir..cs ||Srrr;)r>rrrrcSsdSrrr;rrrrrr )r:Z statementsr2r"inputsgetattrtyper1rZoutputsrr+Z stmt_stackr)r6rrSIRstmtZbefore_stmt_opnumr?Zoutsr)r>rrrun_sirds&     zInterpreter.run_sirrCr cCs$||j}t||}||j|Sr)r:Zsir_name prepare_staterD)r6rCr?rBrrrrcalls  zInterpreter.callcCs|\}}|j|i|Sr)api)r6rCr?argskwargsrrrrGszInterpreter.apicCs.|\}}|d}t||j|ddi|S)Nrr )r@method)r6rCr?rHrIvarrrrrJszInterpreter.methodcCs.|\}}|}|dusJd||i|S)NzSIR bound layer is None.)layer)r6rCr?rHrIrLrrrrLszInterpreter.layerN)r4r)rr8r9r )rr8rr)rCr ) __name__ __module__ __qualname____doc__r7r:rDrFrGrJrLrrrrr3Ps   & r3contextrrr8cstjjfdd}|S)z Compile a SIR to a new function Args: context: The context to compile name: The name of the sir to compile cs(t}|}t||}||S)z{ This function will be decorated by paddle.to_static. so the args is variables, not eager tensors. )r3r:rErD)rH interpreterrBrrQrrrwrappers   zcompile_sir..wrapper)r$ZjitZ not_to_static)rQrrTrrSr compile_sirs  rUcCsNi}t|jrt|j}|r|}t|j|D] \}}|||j<q|Sr)r Zhas_keyrZ get_free_varszipr?)rBr?rZfree_var_seekerZsir_inpinprrrrEs rE)rrrr)rQrrr8) __future__rtypingrr$Z paddle.utilsrutilsrrrZ statement_irr r r r r4rr"r+r&r2r3rUrErrrrs       O