PhddlZddlZddlZddlmZddlmZejeZ eGddZ dejde fd Z dejd efd Zdejd e fd Zd e d efdZd e d efdZy)N) dataclass)configcZeZdZUdZdZeed<dZeed<dedefdZ defdZ dedefd Z y ) CacheSizeRelevantForFramea1 We track the number of cache entries that have same id_match objects as the given frame. TODO(janimesh) - Consider adding a map from tuple_of_match_ids to count - https://github.com/pytorch/pytorch/pull/107496#discussion_r1304564682 - this could be useful for debugging as well. rnum_cache_entriesnum_cache_entries_in_bucketlimitreturncH|j|xs|jSN)will_compilation_exceed_bucketwill_compilation_exceed_totalselfr s cC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/_dynamo/cache_size.pywill_compilation_exceedz1CacheSizeRelevantForFrame.will_compilation_exceedTs&  / / 6 4113 c<|jtjk\Sr )rraccumulated_cache_size_limit)rs rrz7CacheSizeRelevantForFrame.will_compilation_exceed_totalZs%%)L)LLLrc |j|k\Sr )r rs rrz8CacheSizeRelevantForFrame.will_compilation_exceed_bucket]s//588rN) __name__ __module__ __qualname____doc__rint__annotations__r boolrrrrrrrCsUs()( S T MtM9C9D9rrframe local_namec|jj|d}d} tj|}|S#t$rY|SwxYwr )f_localsgetweakrefref TypeError)r r!objweak_ids r_get_weakref_from_f_localsr*asP ..  Z .CG ++c" N   N s7 AAr cddlm}|sy|jj|k7ry|jjj D] \}}|t ||}||k7s yy)z Checks if the ID_MATCH'd objects saved on cache_entry are same as the ones in frame.f_locals, and if the config hash used to compile the cache entry's optimized code is the same as the frame's. r)"get_saved_else_current_config_hashFT) eval_framer,check_fn config_hashid_matched_objsitemsr*)r cache_entryr,r!weakref_from_cache_entryweakref_from_frames r_is_same_cache_bucketr5ksy ? ''+M+OO    - - 3 3 5  # % 1!;E:!N !%== 6 rcpd}d}|r%|dz }t||r|dz }|j}|r%t||S)Nrr)r5nextr)r r2rr s rcompute_cache_sizer8sV"# Q ! 4 '1 , '!&&   %%68S TTr cache_sizec$|jdS)z If the frame (earlier parsed by compute_cache_size) has more than 1 cache entry with same ID_MATCH'd objects, then its a recompilation. r)rr9s ris_recompilationr<s  - -a 00rc@|jtjS)z: Checks if we are exceeding the cache size limit. )rrcache_size_limitr;s rexceeds_cache_size_limitr?s  - -f.E.E FFr)loggingtypesr% dataclassesrr getLoggerrlogr FrameTypestrr*rr5r8r<r?rrrrHs !g!7t 99 9:eoo3$6U ??UU* 1!: 1t 1G)BGtGr