Skip to content

Commit 9b9dc17

Browse files
committed
[AABS] Fix aabs on ppu backend
1 parent 132ee45 commit 9b9dc17

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

python/triton/runtime/adjust_kernel_param.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1337,16 +1337,16 @@ def auto_adjust_block_sizes(nargs, fn, configs, current, config):
13371337
adjust_block_size_dot_m_dim_only(nargs, current, config, tma_m_map, 64) # mthreads
13381338

13391339
if ge_k_map: # tl.dot with general tl.load
1340-
if FLAGTREE_BACKEND == "":
1340+
if FLAGTREE_BACKEND in ("", "ppu"):
13411341
if knobs.autotuning.print:
13421342
print("[AABS] 4. adjust bs in tl.dot with general tl.load")
13431343
adjust_block_size_general_dot_mn_dim(nargs, current, config, ge_k_map, 16)
1344-
if FLAGTREE_BACKEND == "hcu":
1344+
elif FLAGTREE_BACKEND == "hcu":
13451345
if knobs.autotuning.print:
13461346
print("[AABS] 4. adjust bs in tl.dot with general tl.load")
13471347
adjust_block_size_general_dot_mn_dim(nargs, current, config, ge_m_map, 16)
13481348
adjust_block_size_general_dot_mn_dim(nargs, current, config, ge_n_map, 16)
1349-
if FLAGTREE_BACKEND == "sunrise":
1349+
elif FLAGTREE_BACKEND == "sunrise":
13501350
# sunrise min_dot_size = (M=8, N=8, K=16/4) (see sunrise compiler.py
13511351
# min_dot_size). The tl.load shrink path above can lower a BLOCK that
13521352
# also feeds tl.dot below the dot lower bound; bump M/N/K back up to

0 commit comments

Comments
 (0)