7hSSKJr SSKJr SSKJrJrJrJr SSK J r SSK J r J r JrJrJrJr SSKJr SSKJr \"S5r"S S \ \\5r"S S \5rg ))abstractmethod) defaultdict)AnyGenericOptionalTypeVar)override)_serialize_single_cache CacheArtifactCacheArtifactFactoryCacheArtifactManagerCacheArtifactsResult CacheInfo)AppendingByteSerializer) OrderedSetTc\\rSrSrSr\S Sj5r\S\4Sj5r\ S\ 4Sj5r Sr g) PrecompileCacheArtifacta Data for each cache artifact that will be serialized and deserialized by PrecompileContext, rather than CacheArtifactManager. T represents the deserialized type of the artifact, i.e. the return type of after_deserialization PrecompileCacheArtifact is a frozen dataclass - you can add new serializable fields and metadata specific to your own artifacts as needed, and use them in after_deserialization. Example implementation: class MyPrecompileCacheArtifact(PrecompileCacheArtifact[MySerializableType]): my_field: int def after_deserialization(self) -> MySerializableType: result = pickle.loads(self.content) # Do some extra work post deserialization result.my_post_deserialization_function(self.my_field) return result returnNc[S5e)Nz1Precompile cache artifacts do not populate caches) RuntimeErrorselfs Z/var/www/fran/franai/venv/lib/python3.13/site-packages/torch/_dynamo/precompile_context.pypopulate_cache&PrecompileCacheArtifact.populate_cache.sNOOcg)NTrs rprecompile_compatible-PrecompileCacheArtifact.precompile_compatible2srcg)z Code to be run after reading raw byte contents from disk. Generally converts self.content from raw bytes back into its original form. Nr rs rafter_deserialization-PrecompileCacheArtifact.after_deserialization6s rr rN) __name__ __module__ __qualname____firstlineno____doc__r rboolr!rrr$__static_attributes__r rrrrsS(PPt q  rrc ^\rSrSr%Sr0r\\\4\ S'\ "\ 5r \ \ S'\"5r\\\ S'\"\S9r\\\\ \4\ S'\"5r\\ S'\SU4S jj5r\\S \S \S\S S 4Sj55r\SSj5r\S \S \\4Sj5r\S \\\\44U4Sjj5r\ S\ S \4Sj5r!\SSj5r"Sr#U=r$$)PrecompileContext?a PrecompileContext is a special CacheArtifactManager for handling precompilation It uses the same interface as CacheArtifactManager, but handles deserialization differently: instead of placing each artifact into respective caches, it will stitch all the cache artifacts for a single key together and place it into a global Precompile Cache. The following artifact types are supported by PrecompileContext: - BundledAOTAutogradCacheArtifact - CodeStateArtifact (from torch._dynamo.package once available) _new_cache_artifacts_by_key_new_cache_artifacts_seen_artifacts) serialize_fn _serializer _cache_inforNcV>URR5 [TU]5 gN)r1clearsupercls __class__s rr9PrecompileContext.clear]s ''--/  r artifact_typekeycontentc[R"XU5nX@R;agX@RU'URR U5 g)zO Called from each caching operation to record the artifact in this "mega" list N)r encode_creater3r1add)r<r?r@rAartifacts rrecord_artifact!PrecompileContext.record_artifactbsI(55m'R ** * /7'', )rcURR5H9nURURR 5R U5 M; URR 5 g)z We normally record artifacts by key, but serialization expects them to be organized by artifact type. This function transfers artifacts from _new_cache_artifacts_by_key to _new_cache_artifacts N)r1valuesr2r=typeappendr9)r<rEs r_save_artifacts_by_type)PrecompileContext._save_artifacts_by_typezsX 77>>@H  $ $X%7%7%<%<%> ? F Fx PA ''--/rc:URRUS5$)z@ Serialize all artifacts with the given key returned in a list. N)r1get)r<r@s rserialize_artifact_by_key+PrecompileContext.serialize_artifact_by_keys ..223==rc@>UR5 [TU] 5$r8)rLr: serializer;s rrSPrecompileContext.serializes ##%w ""r artifactsc[S5e)NTODO)NotImplementedError)rUs rpopulate_caches!PrecompileContext.populate_cachess !&))rcSSKJn g)Nr)BundledAOTAutogradCacheArtifact)-torch._functorch._aot_autograd.autograd_cacher\)r<r\s r"_ensure_cache_artifacts_registered4PrecompileContext._ensure_cache_artifacts_registereds rr r&)%r'r(r)r*r+r1dictstrr __annotations__rlistr2rrr3rr r5tuplerr6 classmethodr9r rrFrLrrPbytesrS staticmethodrYr^r- __classcell__)r=s@rr/r/?sm  =?c=&8!9>1C>H]4K>> #(5 )9#:;##*#7*I**  rr/N)abcr collectionsrtypingrrrrtyping_extensionsr torch.compiler._cacher r r r rr&torch.utils._appending_byte_serializerrtorch.utils._ordered_setrrrr/r rrrpsY#22&K/ CL# mWQZ# LY ,Y r