Skip to content

Commit a49093e

Browse files
jataylonaromero77amd
authored andcommitted
Update triton_heuristics.py
(cherry picked from commit d5c71f0) (cherry picked from commit 11e1dfc)
1 parent ec5765e commit a49093e

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

torch/_inductor/runtime/triton_heuristics.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2278,8 +2278,7 @@ def foreach(triton_meta, filename=None, inductor_meta=None):
22782278

22792279
# Naive autotuning path for num_warps
22802280
if disable_pointwise_autotuning(inductor_meta) and not (
2281-
inductor_meta.get("max_autotune") or
2282-
inductor_meta.get("max_autotune_pointwise")
2281+
inductor_meta.get("max_autotune") or inductor_meta.get("max_autotune_pointwise")
22832282
):
22842283
configs.append(triton.Config({}, num_stages=1, num_warps=8))
22852284
else:

0 commit comments

Comments
 (0)