diff --git a/torch/_inductor/runtime/triton_heuristics.py b/torch/_inductor/runtime/triton_heuristics.py index 3a39bc677a1c..80efb609a13d 100644 --- a/torch/_inductor/runtime/triton_heuristics.py +++ b/torch/_inductor/runtime/triton_heuristics.py @@ -2419,7 +2419,7 @@ def pointwise( ] if len(size_hints) == 2: if ( - disable_pointwise_autotuning(inductor_meta) or tile_hint == TileHint.SQUARE + disable_pointwise_autotuning(inductor_meta) # or tile_hint == TileHint.SQUARE ) and not ( inductor_meta.get("max_autotune") or inductor_meta.get("max_autotune_pointwise")