ukiUUddlmZddlZddlZddlZddlmZddlZdZde d< ddlZddl m Z ddlmZej eZdZd Zd d ZGd d eZy#e $rY9wxYw)) annotationsN)Anyz Any | Nonefilelockpath)CacheInterfacez-cachez-atimec0|jdxsd|vS)Nzfile://z://) startswithrs M/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/lru_cache.py_is_local_filesystemr 's  # 8uD'88c<eZdZdZddd dZd dZd dZdddd Zy )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_secsct|stjs tdtj|x|_|_|j jdd|dk7|_|jrt td||_ ||_ |j dz |_ t|r%tj|j|_ytj|j|_yy)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)r pathlibepath_installed RuntimeErrorPathr_pathmkdireviction_enabledrmax_sizer lock_pathFileLocklock SoftFileLock)selfrrrs r __init__zLRUCache.__init__6s  %g.E.E { ||$\\$//DI IIOOD4O0$ND   jkkdm0dyy;.dn d #%%dnn5 ))$..9 r c|s td|j|tz }|jr&|jj |j  |jsAtjd| |jr|jjyytjd||j}|jrLtjjdd}|j|tz }|j!|||jr|jjSS#|jr|jjwwxYw)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_SUFFIXrr acquirerexistsloggerdebugrelease read_bytestimetime_nsto_bytes _ATIME_SUFFIX write_bytes)r"key cache_pathval timestamp atime_paths r getz LRUCache.getTsB  , --u]O44J  ii 6 67     +C734      ll(01  ! ! #c  LLN++Ax8 YYC5!88 y)          s(E)BE)E;c|s td|jrMt||jkDr5d|dt|d|jd}t j |y|j |tz }|jr&|jj|j |jr) |jr|jjyy|jt||j||jrLtj j#d d }|j |t$z }|j||jr|jjyy#|jr|jjwwxYw) 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(r))r*rlenrwarningswarnrr+r r,rr-r0_evict_if_neededr6r2r3r4r5)r"r7r9msgr8r:r;s r putz LRUCache.putys~  , -- SDMM!9#C7)CH:>**.--@cmmC u]O44J  ii 6 67          CH5S!  LLN++Ax8 YYC5!88 y)         s1F*BF)F?rr>c|jsyg}d}|jjdtD]}|j }t j s |jn |j}|jjt}|j|tz }tj|jd} ||z }tj || ||f|j"|z } || kDrtj$|\} }}t&j)d||| |j|tz }|j|tz }|j+|j+||z}|| kDryy)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*r)zGEvicting cache entry %r: file size %d bytes, target cache size %d bytes)rrglobr+statrrst_sizelengthname removesuffixr5int from_bytesr1heapqheappushrheappopr.r/unlink) r"r?hdir_sizer8 file_stat file_sizer7r; file_atime target_sizes r rCzLRUCache._evict_if_neededso   %'AHiinnq%89 6 //#i ,3+B+B)## HXHXi OO ( ( 7c99#}o66j>>*"7"7"98Dj)h nnQS)45 6--/1K [ #(==#3 j#y ll014iN99#}o66j99#}o66j)h [ r N)rstrrrNrz float | None)r7rZreturnz bytes | None)r7rZr9bytesr[None)r?rNr[r])__name__ __module__ __qualname____doc__r#r<rErCr r rr+s)UW:<#J(T89-r r)rrZr[bool) __future__rrPloggingr2typingrrAr__annotations__ ImportErrorjax._srcrr$jax._src.compilation_cache_interfacer getLoggerr^r.r+r5r rrbr r rlsx#  *%?   8 $  9e~e#sAA$#A$