diff --git a/aiter/ops/triton/moe/moe_op_gemm_a8w8_blockscale.py b/aiter/ops/triton/moe/moe_op_gemm_a8w8_blockscale.py index 39eb5959d7..e922b425b3 100644 --- a/aiter/ops/triton/moe/moe_op_gemm_a8w8_blockscale.py +++ b/aiter/ops/triton/moe/moe_op_gemm_a8w8_blockscale.py @@ -80,7 +80,7 @@ def get_kernel_config(m, n, k, routing_data): if block_m == 16: block_n = 256 block_k = 128 - num_warps = 4 + num_warps = 8 grid_m = routing_data.n_blocks(m, block_m) grid_n = triton.cdiv(n, block_n)