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

Revert "[Triton] [Inductor] Restrict subprocess autotuning to just Triton (#162688)"

Browse Source 
This reverts commit 082d3dd9d53a60deb022e203892f0c492cf2cce7.

Reverted https://github.com/pytorch/pytorch/pull/162688 on behalf of https://github.com/mlazos due to H100 tests didn't run internally for some reason, rerun with ciflow/h100 ([comment](https://github.com/pytorch/pytorch/pull/162688#issuecomment-3300634763))
This commit is contained in:
PyTorch MergeBot
2025-09-16 23:17:14 +00:00
parent 814338826e
commit e13cf68d03
1 changed files with 2 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
torch/_inductor/select_algorithm.py
View File

@ -3248,13 +3248,8 @@ class AlgorithmSelectorCache(PersistentCache):
# only benchmark triton kernel in sub process for now.
# ATen/Extern kernel are still benchmarked in the current process.
extern = []
triton = []
for c in choices:
if isinstance(c, TritonTemplateCaller):
triton.append(c)
else:
extern.append(c)
extern = [c for c in choices if isinstance(c, ExternKernelCaller)]
triton = [c for c in choices if not isinstance(c, ExternKernelCaller)]
timings = cls.benchmark_in_current_process(
extern, input_nodes, layout, input_gen_fns, hint_override=hint_override