VpfUddlmZddlZddlZddlZddlmZddlZdZde d< ddlZn #e $rYnwxYwddl m Z ddlmZejeZdZd ZddZGddeZdS)) annotationsN)Anyz Any | Nonefilelockpath)CacheInterfacez-cachez-atimerstrreturnboolc4|dpd|vS)Nzfile://z://) startswithrs R/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/lru_cache.py_is_local_filesystemr's  # # 8uD'88c>eZdZdZdddd ZddZddZddddZdS)LRUCachea;Bounded cache with least-recently-used (LRU) eviction policy. This implementation includes cache reading, writing and eviction based on the LRU policy. Notably, when ``max_size`` is set to -1, the cache eviction is disabled, and the LRU cache functions as a normal cache without any size limitations. )lock_timeout_secsrr max_sizeintr float | Nonect|stjstdtj|x|_|_|jdd|dk|_|jrttd||_ ||_ |jdz |_ t|r tj |j |_dStj|j |_dSdS)a_Args: path: The path to the cache directory. max_size: The maximum size of the cache in bytes. Caching will be disabled if this value is set to ``0``. A special value of ``-1`` indicates no limit, allowing the cache size to grow indefinitely. lock_timeout_secs: (optional) The timeout for acquiring a file lock. z`Please install the `etils[epath]` package to specify a cache directory on a non-local filesystemT)parentsexist_okNzMPlease install the `filelock` package to set `jax_compilation_cache_max_size`z .lockfile)rpathlibepath_installed RuntimeErrorPathr_pathmkdireviction_enabledrrr lock_pathFileLocklock SoftFileLock)selfrrrs r__init__zLRUCache.__init__6s  % %}g.E} { | ||$\$///DI IOOD4O000$ND  :  jkkkdm0dy;.dn d # #:%dn55 )$.99  : :rkeyr bytes | Nonec|std|j|tz }|j|tz }|jr |j|j |sBt d| |jr|j dSdSt d|| }tjdd}||||jr|j SS#|jr|j wwxYw)zRetrieves the cached value for the given key. Args: key: The key for which the cache value is retrieved. Returns: The cached data as bytes if available; ``None`` otherwise. key cannot be emptytimeoutzCache miss for key: NzCache hit for key: little) ValueErrorr _CACHE_SUFFIX _ATIME_SUFFIXr"r%acquirerexistsloggerdebugrelease read_bytestimetime_nsto_bytes write_bytes)r'r) cache_path atime_pathval timestamps rgetz LRUCache.getTs . , - --4]444J4]444J 8 i 6777     3C33444    ll000111  ! ! # #c,..))!X66iY'''      s 1E6A.E#E)r@bytesNonecT|std|jrLt||jkr4d|dt|d|jd}t j|dS|j|tz }|j|tz }|jr |j |j  | r% |jr|j dSdS|t|||t!jd d }|||jr|j dSdS#|jr|j wwxYw) zAdds a new entry to the cache. If a cache item with the same key already exists, no action will be taken, even if the value is different. Args: key: The key under which the data will be stored. val: The data to be stored. r,zCache value for key z of size z) bytes exceeds the maximum cache size of z bytesNr-additional_sizer/r0)r1r"lenrwarningswarnrr2r3r%r4rr5r8_evict_if_neededr=r:r;r<)r'r)r@msgr>r?rAs rputz LRUCache.putxs . , - --  SDM!9!9@C@@CHH@@*.-@@@cmC f4]444J4]444J 8 i 6777          CHH555S!!!,..))!X66iY'''     s3F,A4F#F'rrFrGc|jsdSg}d}|jdtD]}|}t js|jn|j}|j t}|j|tz }t |d} ||z }tj|| ||f|j|z } || krtj|\} }}t&d||| |j|tz }|j|tz }||||z}|| kdSdS)a8Evicts the least recently used items from the cache if necessary to ensure the cache does not exceed its maximum size. Args: additional_size: The size of the new entry being added to the cache. This is included to account for the new entry when checking if eviction is needed. Nr*r0zGEvicting cache entry %r: file size %d bytes, target cache size %d bytes)r"rglobr2statrrst_sizelengthname removesuffixr3r from_bytesr9heapqheappushrheappopr6r7unlink) r'rGhdir_sizer> file_stat file_sizer)r? file_atime target_sizes rrKzLRUCache._evict_if_neededs    f%'AHinn%8%8%899 6 6 //##i ,3+BX)## HXi O ( ( 7 7c9#6}666j>>*"7"7"9"98DDj)h nQS)45555-/1K [ #(=#3#3 j#y ll014iNNN9#6}666j9#6}666j)h [ rN)rr rrrr)r)r r r*)r)r r@rCr rD)rGrr rD)__name__ __module__ __qualname____doc__r(rBrMrKrrrr+sUW::::::<""""H''''R89--------rr)rr r r ) __future__rrWloggingr:typingrrIr__annotations__ ImportErrorjax._srcrr$jax._src.compilation_cache_interfacer getLoggerrar6r2r3rrrerrrns)#""""""  ////$%$$$$$??????  8 $ $  9999ccccc~cccccs +33