ȅi9 SSKJr SSKrSSKJrJr SSKJr SSKJ r SSK J r SSK J r SSKJr SS KJr SS KJr SS KJrJrJrJr SS KJrJrJr SS KJr \(aSSKJ r J!r! \"S\"\#\$\S45r%\"S\"\#\$\S4\&\'\\4\(\5r)"SS\*5r+"SS\\\%\)45r,"SS\,\%\)45r-"SS\,\%\)45r."SS\.\%\)45r/"SS\/\%\)45r0g)) annotationsN)ABCabstractmethod) literal_eval)cached_property)sha256)getenv)Path) gettempdir)Lock)AnyGeneric TYPE_CHECKINGTypeVar) assert_neveroverrideSelf)FileLock)FutureThreadPoolExecutorKey.Valuec\rSrSrSrSrg) CacheErrorzB Exception raised for errors encountered during cache operations. N)__name__ __module__ __qualname____firstlineno____doc____static_attributes__rO/home/james-whalen/.local/lib/python3.13/site-packages/torch/_inductor/cache.pyrrsr#rc@\rSrSrSr\SSj5r\SSj5rSrg) Cache$z] Abstract base class for cache implementations. Provides the interface for cache operations. cgz Retrieve a value from the cache. Args: key (Key): The key to look up. Returns: Value | None: The cached value if present, else None. Nr)selfkeys r$get Cache.get*r#cg) Insert a value into the cache. Args: key (Key): The key to insert. value (Value): The value to associate with the key. Returns: bool: True if the value was inserted, False if the key already exists. Nrr*r+values r$insert Cache.insert4r.r#rNr*rr+rreturnz Value | Noner*rr+rr2rr6bool) rrrr r!rr,r3r"rr#r$r&r&$s/     r#r&c^\rSrSrSrS SjrS SjrS Sjr\S Sj5r \SSj5r Sr g ) InMemoryCache@zD In-memory cache implementation using a dictionary and thread lock. c00Ul[5Ulg)z& Initialize an empty in-memory cache. N)_cacher _lockr*s r$__init__InMemoryCache.__init__Es)+ 6 r#cUR URRU5=nb UsSSS5 $SSS5 g!,(df  g=fr))r>r=r,r1s r$r,InMemoryCache.getLs<ZZ--:ZZZs AA AcUR XR;a SSS5 gX RU'SSS5 g!,(df  g=f)r0NFT)r>r=r1s r$r3InMemoryCache.insertYs:ZZkk!Z %KK  ZZs ?? A c U"5n[U5=ncU$URS5HnUR5nU(dMURSS5upVUR5UR5pe[ U5n[ U5n [R"U5n [R"U 5n URX5(aMURU 5U :wdM[ S U <SURU 5<SU <S35e U$![an[ SU<SU<S35UeSnAff=f![[ 4an[ SU<S U<S 35UeSnAff=f![[ 4an[ S U<S U<S 35UeSnAff=f![Ran[ SU<S U<S 35UeSnAff=f![Ran[ S U<S U<S 35UeSnAff=f)a6 Create an in-memory cache from an environment variable. Args: env_var (str): Name of the environment variable containing cache data. Returns: InMemoryCache: An instance populated from the environment variable. Raises: CacheError: If the environment variable is malformed or contains invalid data. N;,zMalformed kv_pair z from env_var z!, likely missing comma separator.zMalformed key_bytes_repr z in kv_pair z, encoding is invalid.zMalformed value_bytes_repr z, not un-pickle-able.zMultiple values for key z found, got z and .) r splitstrip ValueErrorrr SyntaxErrorpickleloadsUnpicklingErrorr3r,) clsenv_varcacheenv_valkv_pairkey_bytes_reprvalue_bytes_reprerr key_bytes value_bytesr+r2s r$ from_env_varInMemoryCache.from_env_varis@g &G /L}}S)GmmoG 3:==a3H0$$& &&(-  (8  *+;<  ll9-   [1LL,,599S>U3J .sg\%))C.ASSXY^Xaabco*v a  ( >'Lmn  ,  //Ag[Xno  ,  12B1E\RYQ\\rs ))  //Ag[Xmn  ))  12B1E\RYQ\\qr sxD: E E-F)G  D=#D88D=E*E%%E*-F=FFG.GG G9G44G9czU"5nUR5(dU$[US5n[R"U5UlSSS5 [UR[5(d[ SUS35eU$!,(df  N>=f![R an[ SUS35UeSnAff=f)z Create an in-memory cache from a file path. Args: fpath (Path): Path to the file containing pickled cache data. Returns: InMemoryCache: An instance populated from the file. Raises: CacheError: If the file is not a valid pickled dictionary. rbNz&Failed to create cache from file path z#, file contents are un-pickle-able.z-, file contents not pickled dict[Key, Value].) is_fileopenrOloadr=rQr isinstancedict)rRfpathrTfprYs r$from_file_pathInMemoryCache.from_file_paths}}L eT"b%{{2 #%,,--8?lm  #"%% 8?bc  s4 BBB B BBB:%B55B:)r=r>Nr*rr6Noner5r7)rSstrr6r)rer r6r) rrrr r!r@r,r3 classmethodr\rgr"rr#r$r:r:@sA"  KKZr#r:cP\rSrSrSrSSjrSSjrSrg) AsyncCachez= Asynchronous cache implementation using ThreadPoolExecutor. c:URURU5$)z Retrieve a value from the cache asynchronously. Args: key (Key): The key to look up. executor (ThreadPoolExecutor): Executor for async execution. Returns: Future[Value | None]: Future for the cached value or None. )submitr,)r*r+executors r$ get_asyncAsyncCache.get_asyncstxx--r#c:URURX5$)a Insert a value into the cache asynchronously. Args: key (Key): The key to insert. value (Value): The value to associate with the key. executor (ThreadPoolExecutor): Executor for async execution. Returns: Future[bool]: Future for the result of insertion. )rqr3)r*r+r2rrs r$ insert_asyncAsyncCache.insert_asyncst{{C77r#rN)r*rr+rrrrr6zFuture[Value | None]) r*rr+rr2rrrrr6z Future[bool])rrrr r!rsrvr"rr#r$rnrnsY . . .(: .  . 8 8 8%* 86H 8  8r#rnc\rSrSr%SrSrS\S'SSSjjr\SSj5r SS jr SS jr \ SS j5r \SS j5r\SS j5rSrg) OnDiskCachea# On-disk cache implementation using files and file locks. Stores cache data in files on disk, with atomic operations and versioning. Supports custom cache directory names. Attributes: version (int): The version used for cache versioning. name (str): The name of the cache directory. rintversionNc$U=(d SUlg)z Initialize an on-disk cache instance. Args: name (str | None, optional): The name of the cache directory. If None, defaults to "on_disk_cache". on_disk_cacheNname)r*rs r$r@OnDiskCache.__init__s+O r#cH[[55S- UR- $)zk Get the base directory for the cache. Returns: Path: The base directory path for storing cache files. rT)r r rr?s r$base_dirOnDiskCache.base_dirsJL!G+dii77r#cUR[[R"U55R 5SS- $![ [R 4an[SU<S35UeSnAff=f)z Get the file path for a given key. Args: key (Key): The key to convert to a file path. Returns: Path: The file path for the key. Raises: CacheError: If the key is not pickle-able. N zFailed to get fpath for key z, key is not pickle-able.) rrrOdumps hexdigestAttributeError PicklingErrorrr)r*r+rYs r$_fpath_from_keyOnDiskCache._fpath_from_keysl ==6&,,s*;#<#F#F#H"#MM M 4 45 .sg5NO  s Retrieve a value from the cache. Args: key (Key): The key to look up. Returns: Value | None: The cached value if present and version matches, else None. Raises: CacheError: If the value is corrupted or cannot be unpickled. Side Effects: Removes stale cache files if the version prefix does not match. Nr_zFailed to get key z?, value is potentially corrupted (value is not un-pickle-able).) rrr`lenrrareadunlinkrOrPrQr) r*r+reflockr[ prefix_lengthrfr2rYs r$r,OnDiskCache.getBs$$S)&&u- ==??UK 3 34MeT"b77=)T-@-@@"$'')K#" U"  [1 1U #"))  (/no 'UsMD#D(0CD>C0D C- )D0DDDD D+cURU5nURU5nURRSSS9 Un[ US5nUR UR 5 [R"X&5 SSS5 SSS5 g!,(df  N=f!,(df  g=f![Ran[SU<SU<S35UeSnAf[a gf=f) aQ Insert a value into the cache. Args: key (Key): The key to insert. value (Value): The value to associate with the key. Returns: bool: True if the value was inserted, False if the key already exists. Raises: CacheError: If the value is not pickle-able. Side Effects: Creates the cache directory if it does not exist. T)parentsexist_okxbNzFailed to insert key z with value z, value is not pickle-able.F) rrrmkdirrawriterrOdumprrFileExistsError)r*r+r2rer_rfrYs r$r3OnDiskCache.insertls$$S)&&u- 4$7 !T%.",,- E&//. ## 'wl5)C^_   sSB2 B! 2B?B!B2 B B!! B/+B2/B22C+C C+*C+r)N)r*rrz str | Noner6rjr*rr6r )r*rr+rr6r )r*rrer r6r)r*rr6bytesr5r7)rrrr r!r|__annotations__r@rrrrpropertyrrr,r3r"rr#r$ryrysvGS,88& PFF''Rr#rycD^\rSrSrSrSU4Sjjr\SSj5rSrU=r $)InductorOnDiskCacheizh Inductor-specific on-disk cache implementation. Uses a custom base directory for Inductor cache files. c$>[TU]S5 g)zk Initialize an inductor on-disk cache instance. Sets the cache directory name to "inductor_on_disk_cache". inductor_on_disk_cacheN)superr@)r* __class__s r$r@InductorOnDiskCache.__init__s 12r#cFSSKJn [U"5SUR5$)zu Get the base directory for the Inductor cache. Returns: Path: The base directory path for Inductor cache files. r)default_cache_dirrT)%torch._inductor.runtime.runtime_utilsrr r)r*rs r$rInductorOnDiskCache.base_dirs L%'$))<rs" #%77::*=eS#uS#X/c5c?E4S>4PS9U  Ce$ 8UE#u*%Up8sEz"8BQ*S%Z(Qh=+c5j1=r#