^j ddlmZddlZddlZddlmZddlmZmZm Z ddl m Z er ddl m Z ddlmZe dZe d Ze d ZeZed Zdd Z ddZy)) annotationsN)ThreadPoolExecutor) Concatenate TYPE_CHECKINGTypeVar) ParamSpec)Callable)Future_P_T_C) max_workerscttjddfd }dfd }dfd }dfd }||_||_||_|S) z Like functools.cache but with a prefetch() method that starts computing the value in a background thread, and a set() method for prepopulating. Nc5tur cdddSjdcdddScdddS#1swYyxYwN)_cache_sentinelresult_cache_future_lockfuncsa/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/utils/_functools.pywrapperz#prefetchable_cache..wrapper"sb  _,  " )   VF   s A A A  Ac^5tur td|dddy#1swYyxYw)Nz$prefetchable_cache value already set)r RuntimeError)valrrs rset_valz#prefetchable_cache..set_val.s4  _,"#IJJF   s#,cD5tddddy#1swYyxYwr)r)rrrsrclearz!prefetchable_cache..clear5s%  $FG   s c5tus dddytjdddy#1swYyxYwr)r_prefetch_executorsubmitrsrprefetchz$prefetchable_cache..prefetch;sG  6_,0C 6 6)//5G 6 6 6s 77A)returnr )rr r&None)r&r')r threadingLocksetr!r%)rrrr!r%rrrs` @@@rprefetchable_cacher+sV *F NN E!%G   66GKGMG Ncfdjztjdfd }|S)a Like `@functools.cache` but for methods. `@functools.cache` (and similarly `@functools.lru_cache`) shouldn't be used on methods because it caches `self`, keeping it alive forever. `@cache_method` ignores `self` so won't keep `self` alive (assuming no cycles with `self` in the parameters). Footgun warning: This decorator completely ignores self's properties so only use it when you know that self is frozen or won't change in a meaningful way (such as the wrapped function being pure). _cache_method_c|r tdt|dx}si}t|||j|t}|tur|S|g|i|}|||<|S)Nz.cache_method does not accept keyword arguments)AssertionErrorgetattrsetattrgetr)selfargskwargscache cached_valuevalue cache_namefs rwrapzcache_method..wrapYsw  !QR R z4888E D*e ,yy7  . $(((d  r,)r4r r5z_P.argsr6z _P.kwargsr&r )__name__ functoolswraps)r;r<r:s` @r cache_methodr@Hs5"AJJ.J__Q   Kr,)rCallable[[], _T]r&rA)r;!Callable[Concatenate[_C, _P], _T]r&rB) __future__rr>r(concurrent.futuresrtypingrrrtyping_extensionsrcollections.abcr r r r r objectrr#r+r@r,rrJsq"166'()t_ T] T]('A6,^(&r,