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:
PyTorch MergeBot
2025-07-21 20:45:21 +00:00
parent b1a0c34dd3
commit bc379aebe2
4 changed files with 3 additions and 79 deletions

View File

@ -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