mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Revert "Still run TritonBundler with BundledAOTAutogradCache, save autotune results (#158048)"
This reverts commit 8e57cdb746b4ab28865fdf01532f87b0d21700e9. Reverted https://github.com/pytorch/pytorch/pull/158048 on behalf of https://github.com/jeffdaily due to rocm failures due to unit test introduced in this PR, but no pre-merge signal available ([comment](https://github.com/pytorch/pytorch/pull/158048#issuecomment-3098746624))
This commit is contained in:
@ -70,8 +70,7 @@ class PrecompileContext(CacheArtifactManager):
|
||||
|
||||
The following artifact types are supported by PrecompileContext:
|
||||
- BundledAOTAutogradCacheArtifact
|
||||
- DynamoCodeStateArtifact
|
||||
- AutotuneCacheArtifact (regular autotune results, same as Megacache)
|
||||
- CodeStateArtifact (from torch._dynamo.package once available)
|
||||
"""
|
||||
|
||||
# Protected by the compile_lock
|
||||
@ -150,12 +149,8 @@ class PrecompileContext(CacheArtifactManager):
|
||||
artifacts_by_key = {}
|
||||
cache_info = CacheInfo()
|
||||
for artifact in chain(*artifacts.values()):
|
||||
if artifact.type() == "autotune":
|
||||
# Populate autotune cache artifacts
|
||||
artifact.populate_cache()
|
||||
else:
|
||||
artifacts_by_key[artifact.key] = artifact
|
||||
cache_info.add(artifact)
|
||||
artifacts_by_key[artifact.key] = artifact
|
||||
|
||||
from torch._dynamo.package import _BackendId, DynamoCache
|
||||
|
||||
|
Reference in New Issue
Block a user