diff --git a/dsv4/kernels/attention/fmha_6warp_tma_multitile.cuh b/dsv4/kernels/attention/fmha_6warp_tma_multitile.cuh index f8f29fb4..c9b87dcc 100644 --- a/dsv4/kernels/attention/fmha_6warp_tma_multitile.cuh +++ b/dsv4/kernels/attention/fmha_6warp_tma_multitile.cuh @@ -201,7 +201,7 @@ fmha_6warp_tma_multitile_kernel(FmhaTmaMultiTileParams params) { s_vals[j] = expf(s_vals[j] - row_max); row_sum += s_vals[j]; } - for (int j=0;j