o "j@sddlmZddlZddlmZddlZddlmZddlm Z ddl m Z ddl m Z dd lmZmZmZmZmZmZmZd d lmZerLd d lmZd dZddZGdddZeGdddeZdS)) annotationsN) TYPE_CHECKING) amp_state) convert_dtype)_dygraph_tracer) EventGuard)Cache GraphLogger SingletonStepInfoManagerloglog_domap_if) compile_sir)SymbolicTraceContextcCs>t}|jdur|j}|j}tjj||jdus dSdSN)inspect currentframef_backf_codepaddleZ frameworkcoreZsot_set_with_graph)framecoderf/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/jit/sot/symbolic/compile_cache.pytrace_back_frames)s  rcCs|D]}d|_qdS)N)name)Zoutput_tensorsZ output_tensorrrrclear_eager_tensor_name1sr!c@s*eZdZdZd ddZddZdd Zd S) FallbackWrapperzW Used to store and call static graph methods generated by paddle.jit.to_static is_trainingboolcCs"||_d|_d|_||_||_dSr) compiled_fnpartial_programconcrete_programSIRr#)selfr%r(r#rrr__init__;s  zFallbackWrapper.__init__cst}|dur ||fSt}|js!|js!|js!||fSt|dtdddt||ffddddd dd S) zSs z1FallbackWrapper.amp_cast_inputs..cSs |tjSr)castrZfloat32r.rrrr1U cSs|Srrr.rrrr1Vs)predZtrue_fnZfalse_fn) rrZ_expected_placeZ is_gpu_placeZ is_xpu_placeZis_custom_placerr r)r)argskwargsZcurrent_amp_stateZtracerrr0ramp_cast_inputsBs&  zFallbackWrapper.amp_cast_inputsc s8tdjjtjrttdfdd\tdfddjdurXtdj j i\_ _j j_ Wdn1sSwYtdjji}Wdn1spwYt|td fd dtdd d|WdS1swYdS) NzFallbackWrapper: rcs tdjS)Nz"[FallbackWrapper] start run SIR: )printr(rr)rrr1`r3z*FallbackWrapper.__call__..cstjjidjS)Nr)r8r%get_concrete_programZ train_programrr5r6r)rrr1es z%FallbackWrapper: get_concrete_programz)FallbackWrapper: sot call partial_programrcstjjSr)r Z add_subgraphr'Z main_programrr9rrr1xscSstdS)Nz'[CompileCache] run sir forward success.)r8rrrrr1~s)rr(r r Zneed_back_tracerrr7r&r%r;r'r#trainingZsot_callr!)r)r5r6Zoutputsrr<r__call__YsD      $zFallbackWrapper.__call__N)r#r$)__name__ __module__ __qualname____doc__r*r7r>rrrrr"6s   r"cs4eZdZdZfddZd dd Zd d d ZZS) CompileSIRCachez, Cache the compiled function of SIR cstjdddS)NF)Zweak)superr*r9 __class__rrr*szCompileSIRCache.__init__contextrsir_namestrcKs"||}tt||df}|S)a generate a hash key for a SIR Args: context: The context to compile sir_name: The name of the sir to compile build_strategy: The build strategy to compile Returns: The hash key of the SIR r=)get_sirhashrI)r)rGrHr6ZsirZhash_keyrrrkey_fns zCompileSIRCache.key_fncKsF|dd}|dd}ttjjt||||dd|||ddS)a Generate static graph function Args: context: The context to compile sir_name: The name of the sir to compile build_strategy: The build strategy to compile Returns: The static graph function build_strategyNbackendT)rMrNZ full_graphr=)r#)getr"rZjitZ to_staticrrJ)r)rGrHr6rMrNrrrvalue_fns zCompileSIRCache.value_fn)rGrrHrI)r?r@rArBr*rLrP __classcell__rrrErrCs   rC) __future__rrtypingrrZpaddle.amp.auto_castrZpaddle.base.data_feederrZpaddle.frameworkrZprofilerrutilsr r r r r rr interpreterrZsymbolic_contextrrr!r"rCrrrrs"      $  M