frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Introduce remote cache key prefix to break cache (#142148)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/142148
Approved by: https://github.com/jamesjwu, https://github.com/ezyang
This commit is contained in:
Oguz Ulgen
2024-12-06 14:31:14 -08:00
committed by PyTorch MergeBot
parent 1cb5f38328
commit 0f6bfc58a2
7 changed files with 125 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/_inductor/codecache.py
View File

@ -66,6 +66,7 @@ from torch._inductor.codegen.rocm.compile_command import (
from torch._inductor.custom_graph_pass import CustomGraphPass, CustomGraphPassType
from torch._inductor.output_code import has_frozen_params
from torch._utils_internal import log_cache_bypass
from torch.compiler import config as cconfig
from .remote_cache import create_cache
from .runtime import autotune_cache
@ -782,6 +783,7 @@ class FxGraphHashDetails:
) -> None:
self.gm = gm
self.example_inputs = example_inputs
self.cache_key_tag = cconfig.cache_key_tag
# Order kwargs so hashing is stable to changes in kwarg order. Although
# it's technically a _CompileFxKwargs we don't actually need it typed as