Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
grimoire committed Nov 24, 2023
1 parent 295715d commit a81bde3
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ __global__ void flash_fwd_kernel(Flash_fwd_params params)
#if __CUDA_ARCH__ >= 800
flash::compute_attn<Kernel_traits, Is_dropout, Is_causal, Is_even_N, Is_even_K, Return_softmax>(params);
#else
// TODO: support flash attention2 on sm<80
// TODO: support flash attention2 on sm<80
assert(false);
#endif
}
Expand Down

0 comments on commit a81bde3

Please sign in to comment.