diff --git a/torch/_inductor/codegen/multi_kernel.py b/torch/_inductor/codegen/multi_kernel.py index bff05a5878a9..bb42d340333f 100644 --- a/torch/_inductor/codegen/multi_kernel.py +++ b/torch/_inductor/codegen/multi_kernel.py @@ -313,7 +313,14 @@ class MultiKernelCall: self._recorded = False def cache_file_path(self): - key = code_hash(",".join([k.fn.cache_key for k in self.kernels])) + key = code_hash( + ",".join( + [ + f"{k.fn.cache_key}{k.size_hints!r}{k.triton_meta!r}" + for k in self.kernels + ] + ) + ) _, _, path = get_path(key, "picked_kernel") return pathlib.Path(path)