o "jI@sZddlmZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZmZmZmZmZddlmZddlZddlZddlmZdd lmZmZd d lmZmZmZm Z d d l!m"Z"m#Z#m$Z$ed Z%Gdddee%Z&GdddZ'da(GdddZ)e ddZ*ddZ+e&GdddZ,ddZ-ddZ.ddZ/d d!Z0d"d#Z1d$d%Z2d&d'Z3d(d)Z4d*d+Z5d,d-Z6d.d/Z7d0d1Z8Gd2d3d3Z9d4d5Z:d6d7Z;d8d9ZdXdEdFZ?dGdHZ@e&GdIdJdJZAe&GdKdLdLZBdMdNZCGdOdPdPee%ZDGdQdRdRe ZEGdSdTdTZFe&GdUdVdVZGdS)Y) annotationsN) OrderedDict)contextmanager)Enum)AnyGenericIterableIteratorTypeVar)WeakValueDictionary)Program)flatten map_structure)ENV_CLEAN_CODEENV_COST_MODELENV_SOT_LOG_LEVELENV_STRICT_MODE)break_graph_setpaddle_api_listpaddle_api_module_prefixTc@s eZdZd ddZd ddZd S) Singletonclstype[T]cCs||_i|_dSN_clsZ _instance)selfrr[/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/jit/sot/utils/utils.py__init__2 zSingleton.__init__returnrcCs(|j|jvr||j|j<|j|jSrrrrrr __call__6s  zSingleton.__call__N)rr)r#r)__name__ __module__ __qualname__r!r%rrrr r1s rc@s&eZdZddZddZd d d Zd S) NameGeneratorcCsd|_||_dSNr)counterprefix)rr,rrr r!=r"zNameGenerator.__init__cCs"|jt|j}|jd7_|SNr)r,strr+rnamerrr nextAszNameGenerator.nextr0r.r#boolcCs ||jSr) startswithr,r/rrr match_nameF zNameGenerator.match_nameN)r0r.r#r2)r&r'r(r!r1r4rrrr r)<sr)c@s$eZdZddZddZddZdS)TmpNameRecordscCstdd|_t|_dS)NZ _sot_tmp_)r,)r)name_generatorrtmp_names_recordr$rrr r!Ns  zTmpNameRecords.__init__cC |jSr)r7r1r$rrr next_nameRs zTmpNameRecords.next_namecCs*||jvr |j|S|}||j|<|Sr)r8r:)rexprtmp_namerrr add_tmp_varUs   zTmpNameRecords.add_tmp_varN)r&r'r(r!r:r=rrrr r6Ms r6ccst}tadV|adSr)_tmp_name_recordsr6)oldrrr tmp_name_guard^s r@cCstSr)r>rrrr current_tmp_name_recordsgsrAc@seZdZdddZddZdS) ResumeFnNameFactoryr#NonecCstd|_dS)NZresume_)r)genr$rrr r!nszResumeFnNameFactory.__init__cCs|j}|Sr)rDr1r/rrr r1qs zResumeFnNameFactory.nextN)r#rC)r&r'r(r!r1rrrr rBls  rBcGs&t}||krt|ddidSdS)Nend)rgetprint)levelargs cur_levelrrr logvsrLcCst}||kr |dSdSr)rrG)rIfnrKrrr log_do|s rNcGs*t}||krt|j|dddSdS)NrF)rE)rrGrHformat)rIr.rJrKrrr log_formatsrPcfdd}|S)Nc sPtjjd}zz |i|}WnWtjj||Stjj|wr)paddleZ frameworkcoreZset_eval_frame)rJkwargsZold_cbretvalfuncrr no_eval_frame_funcsz)no_eval_frame..no_eval_frame_funcr)rWrXrrVr no_eval_frames  rYcCs>t|tjjr dSt|drdSt|rdSt|p|tvS)NF__self__) isinstancerRnnZLayerhasattrinspectisclassin_paddle_modulerrVrrr is_paddle_apis racCsTtjg}||vr dSt|tjrdSttD]\}}||ur't|tr'dSqdSNTF) weakrefrefr[typesBuiltinFunctionTyper^ getmembersbuiltinstype)rMZspecial_builtin_fns member_namememberrrr is_builtin_fns rlcCst|dr9|j}|durdStdd|dt|dr"tdd|jdtddtD]}||r8tdd d Sq)tdd dS) Nr'Fz&find paddle function with __module__:  r&z& with __name__ : z& with results : z True Tz False )r]r'rLr&rr3)rWZ module_strr,rrr r`s"       r`cCs|tvSr)rrVrrr is_break_graph_apirocs fdd}t|g|RS)Ncs|r|S|SrrrJfalse_fnpredtrue_fnrr replaceszmap_if..replace)r)rtrurs structuresrvrrrr map_ifsrxccs>t|D]}t|tr|jV|jV|jVq|VqdSr)r r[slicestartstopstep) structureitemrrr flatten_extends   rcs,fdd}fdd}t|||dS)z8support extended structures like slice and SliceVariablecst|trdS|S)NT)r[ry)xrtrr wrapped_preds z#map_if_extend..wrapped_predcs8t|tr|j|j|jg}t|}t|S|Sr)r[ryrzr{r| map_if_extend)rlrrrr wrapped_true_fns z&map_if_extend..wrapped_true_fn)rtrurs)rx)r}rtrursrrrrrr rs rcs$fdd}ttt|g|RS)Ncs|rdSdSNrrrrqrrr is_trueszcount_if..is_true)sumr r)rtrwrrrr count_ifs rc@s6eZdZd ddZddZddZdd Zd d Zd S)CacheFcCs|si|_nt|_d|_dSr*)cacher hit_num)rZweakrrr r!s zCache.__init__cOsv|j|i|}|dur|j|i|S||jvr,tdd|d|jd7_|j|S|j|i|}||j|<|S)Nrmz cache hit: rnr)key_fnvalue_fnrrLr)rrJrT cache_keyvaluerrr r%s   zCache.__call__cCs|jd|_dSr*)rclearrr$rrr r s  z Cache.clearcOtrNotImplementedErrorrrJrTrrr r z Cache.key_fncOrrrrrrr rrzCache.value_fnN)F)r&r'r(r!r%rrrrrrr rs    rcrQ)Ncs4t}|i|}t}||}td||S)Nz Execute time:)timerH)rJrT start_timeresultZend_timeZexecution_timerVrr wrappers  zexecute_time..wrapperr)rWrrrVr execute_times rcCsd|d|d|dS)Nz(shape: z , dtype: z, stop_gradient: )r)shapeZdtypeZ stop_gradientrrr meta_str rcCtSr)rrGrrrr is_strict_mode$rprr#r2cCrr)rrGrrrr is_clean_code(rprli list[Any]r~rintcCsdd|Dt|S)NcSg|]}t|qSrid.0itrrr -z)list_find_index_by_id..)indexrrr~rrr list_find_index_by_id,rrcCst|dd|DvS)NcSrrrrrrr r1rz&list_contain_by_id..rrrrr list_contain_by_id0srcCs t|j|Sr)getattr __class__)objr0rrr get_unbound_method4s rc@sxeZdZUded<ded<ded<ded<dd Zd d Zd d ZddZdddZddZ ddZ ddZ ddZ dS) GraphLoggerr graph_numop_numz list[Program]graphsz$list[paddle.base.framework.Operator]opscCs |dSr)rr$rrr r!@r5zGraphLogger.__init__cCsd|_d|_g|_g|_dSr*)rrrrr$rrr rCs zGraphLogger.clearcC|jSr)rr$rrr get_graph_numIrzGraphLogger.get_graph_numcCrr)rr$rrr get_op_numLrzGraphLogger.get_op_numprogramr cCsT|jd7_|j|d}|jD] }|jd7_|d7}q|j|dSr)rrappendZ global_blockrr)rrZ sub_op_numoprrr add_subgraphOs  zGraphLogger.add_subgraphcCsPtt|jD]}|d|d|d|j|||j|qdS)Nz6------------------------------------------------------z subgraph z , OpNum: )rangelenrrr)rstrsirrr add_subgprah_infoYszGraphLogger.add_subgprah_infocs^gdddtdfddddS)Nz6---------------- PaddleSOT graph info ----------------z SubgraphNum: zOpNum: rmcs Sr)rrrrrr is z%GraphLogger.__str__..rn)rrrrNjoinr$rrr __str__bs   zGraphLogger.__str__cCs|Sr)rr$rrr __repr__nrpzGraphLogger.__repr__cCs t|dSr)rHr$rrr print_infoqr5zGraphLogger.print_infoN)rr ) r&r'r(__annotations__r!rrrrrrrrrrrr r9s     rc@s eZdZdS)SotUndefinedVarN)r&r'r(rrrr rusrc Cs6zt|WdSty}zWYd}~dSd}~wwrb)hash TypeError)rerrr hashablezsrc@seZdZUdZded<d/d0ddZd1d d Zd2ddZd3ddZd2ddZ d3ddZ d2ddZ d3ddZ d4ddZ d4dd Zd5d"d#Zd6d%d&Zd7d'd(Zd8d*d+Zd9d-d.ZdS): OrderedSetz6 A set that preserves the order of insertion. z dict[T, None]_dataNitemsIterable[T] | NonecCs"|dur t||_dSi|_dS)z Examples: >>> s = OrderedSet([1, 2, 3]) >>> s OrderedSet(1, 2, 3) >>> s = OrderedSet() >>> s OrderedSet() N)dictfromkeysr)rrrrr r!s" zOrderedSet.__init__r# Iterator[T]cC t|jS)z Examples: >>> s = OrderedSet([1, 2, 3]) >>> for item in s: ... print(item) 1 2 3 )iterrr$rrr __iter__ zOrderedSet.__iter__other OrderedSet[T]cCstt|t|S)a7 Union two sets. Args: other: Another set to be unioned. Returns: The union of two sets. Examples: >>> s1 = OrderedSet([1, 2, 3]) >>> s2 = OrderedSet([2, 3, 4]) >>> s1 | s2 OrderedSet(1, 2, 3, 4) )rlistrrrrr __or__szOrderedSet.__or__cCs|jt||S)a Union two sets in place. Args: other: Another set to be unioned. Examples: >>> s1 = OrderedSet([1, 2, 3]) >>> s2 = OrderedSet([2, 3, 4]) >>> s1 |= s2 >>> s1 OrderedSet(1, 2, 3, 4) )rupdaterrrrrr __ior__szOrderedSet.__ior__ctfdd|DS)a@ Intersect two sets. Args: other: Another set to be intersected. Returns: The intersection of two sets. Examples: >>> s1 = OrderedSet([1, 2, 3]) >>> s2 = OrderedSet([2, 3, 4]) >>> s1 & s2 OrderedSet(2, 3) csg|]}|vr|qSrrrr~rrr rz&OrderedSet.__and__..rrrrr __and__zOrderedSet.__and__cfdd|D|_|S)a! Intersect two sets in place. Args: other: Another set to be intersected. Examples: >>> s1 = OrderedSet([1, 2, 3]) >>> s2 = OrderedSet([2, 3, 4]) >>> s1 &= s2 >>> s1 OrderedSet(2, 3) csi|] }|vr|dqSrrrrrr z'OrderedSet.__iand__..rrrrr __iand__zOrderedSet.__iand__cr)a: Subtract two sets. Args: other: Another set to be subtracted. Returns: The subtraction of two sets. Examples: >>> s1 = OrderedSet([1, 2, 3]) >>> s2 = OrderedSet([2, 3, 4]) >>> s1 - s2 OrderedSet(1) csg|]}|vr|qSrrrrrr rrz&OrderedSet.__sub__..rrrrr __sub__rzOrderedSet.__sub__cr)a Subtract two sets in place. Args: other: Another set to be subtracted. Examples: >>> s1 = OrderedSet([1, 2, 3]) >>> s2 = OrderedSet([2, 3, 4]) >>> s1 -= s2 >>> s1 OrderedSet(1) csi|] }|vr|dqSrrrrrr rrz'OrderedSet.__isub__..rrrrr __isub__rzOrderedSet.__isub__r~rcCs|j|dS)z Add an item to the set. Args: item: The item to be added. Examples: >>> s = OrderedSet([1, 2, 3]) >>> s.add(4) >>> s OrderedSet(1, 2, 3, 4) N)r setdefaultrr~rrr add s zOrderedSet.addcCs |j|=dS)z Remove an item from the set. Args: item: The item to be removed. Examples: >>> s = OrderedSet([1, 2, 3]) >>> s.remove(2) >>> s OrderedSet(1, 3) Nrrrrr removes zOrderedSet.remover2cCs ||jvS)z Examples: >>> s = OrderedSet([1, 2, 3]) >>> 1 in s True >>> 4 in s False rrrrr __contains__(s zOrderedSet.__contains__rcCr)zj Examples: >>> s = OrderedSet([1, 2, 3]) >>> len(s) 3 )rrr$rrr __len__3s zOrderedSet.__len__cCr)z Examples: >>> s = OrderedSet([1, 2, 3]) >>> bool(s) True >>> s = OrderedSet() >>> bool(s) False )r2rr$rrr __bool__<rzOrderedSet.__bool__objectcCst|tstSt|t|kS)z Examples: >>> s1 = OrderedSet([1, 2, 3]) >>> s2 = OrderedSet([1, 2, 3]) >>> s1 == s2 True >>> s3 = OrderedSet([3, 2, 1]) >>> s1 == s3 False )r[rNotImplementedrrrrr __eq__Hs zOrderedSet.__eq__r.cCsdtt|j}d|dS)Nz, z OrderedSet(r)rmapreprr)rZ data_reprrrr rWs zOrderedSet.__repr__r)rr)r#r)rrr#r)rr)r~r)r~rr#r2)r#rr#r2)rrr#r2)r#r.)r&r'r(__doc__rr!rrrrrrrrrrrrrrrrrr rs$           rc@seZdZdZdZdZdS) StepStaterN)r&r'r( COLLECT_INFORUN_SOTRUN_DYNrrrr r\src@sLeZdZdZdZdZdZdZdZddZ dd Z d d Z d d Z ddZ dS)StepInfo rm2g?cCs8d|_tr tjntj|_g|_d|_g|_ d|_ dS)Nr) step_countrrGrrrstatedyn_time_costs avg_dyn_timesot_time_costssot_stepr$rrr r!ms zStepInfo.__init__cCs>|j|t|j|jkrt|j|j d|_dSdSr)rrrREQUIRED_DYN_INFOSnpmeanUSED_DYN_INFOSr)r time_costrrr add_dynamic_time_infoys  zStepInfo.add_dynamic_time_infocCs|j|t|j|jkret|j}tdd|d|jd||jkr5tdd|dtj |_ dS|j |j ksFt |j||jkrUtdd|dtj|_ dStdd|d|jdSdS) Nrz[Cost Model] sot: z, dyn: rnz [Cost Model] Switch to RUN_SOT: z z [Cost Model] Switch to RUN_DYN: z[Cost Model] Decision delayed: )rrrREQUIRED_SOT_INFOSrrrLrrrrr COLLECT_INFO_MAX_STEPZstd CV_BOUNDARYrr)rr current_codeZ avg_sot_timerrr add_sot_time_infos"      zStepInfo.add_sot_time_infocCs |j|jkSr)r BACK_TRACE_STEPSr$rrr need_back_tracer5zStepInfo.need_back_tracecCst|j|jkSr)rrrr$rrr need_dynamic_infoszStepInfo.need_dynamic_infoN)r&r'r(rrrrrrr!rrrr rrrr rbs  rc@s\eZdZddZeddZddZddZed d Z ed d Z ed dZ ddZ dS)StepInfoManagercCsi|_d|_d|_dSr) step_recordrcurrent_step_infor$rrr r!s zStepInfoManager.__init__ccsz8|j}|j}||_||jvrt|j|<|j||_|jjd7_tdd|jddVW||_||_dS||_||_w)Nrrz.[Cost Model] New step start, current state is rn)rr#r"rr rL current_state)rcodeZold_codeZold_inforrr step_guards$     zStepInfoManager.step_guardcCs|jjd7_dSr-)r#rr$rrr rszStepInfoManager.sot_stepcOsr|jrt}||i|}t|}|j||St}||i|}t|}|j||j|Sr)r#r r perf_counterrrr)rZ impl_dynamicZimpl_sotrJrTrZoutsrrrr collect_infos    zStepInfoManager.collect_infocCr9r)r#rr$rrr rs zStepInfoManager.need_back_tracecC|jjSr)r#r r$rrr current_stepzStepInfoManager.current_stepcCr)r)r#rr$rrr r$r+zStepInfoManager.current_statecCs|jd|_d|_dS)Nr )r"rrr*r$rrr rs  zStepInfoManager.clearN) r&r'r(r!rr&rr(propertyrr*r$rrrrr r!s     r!r)rrr~rr#r)H __future__rrhr^rrerc collectionsr contextlibrenumrtypingrrrr r r numpyrrRZpaddle.frameworkr Z paddle.utilsr rZenvsrrrrZpaddle_api_configrrrrrr)r>r6r@rArBrLrNrPrYrarlr`rorxrrrrrrrrrrrrrrrrrr!rrrr st                  ;[: