@@ -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