diff options
Diffstat (limited to 'candle-flash-attn/kernels/flash_fwd_hdim256_fp16_sm80.cu')
| -rw-r--r-- | candle-flash-attn/kernels/flash_fwd_hdim256_fp16_sm80.cu | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-flash-attn/kernels/flash_fwd_hdim256_fp16_sm80.cu b/candle-flash-attn/kernels/flash_fwd_hdim256_fp16_sm80.cu index 9f35129c..9c9a1715 100644 --- a/candle-flash-attn/kernels/flash_fwd_hdim256_fp16_sm80.cu +++ b/candle-flash-attn/kernels/flash_fwd_hdim256_fp16_sm80.cu @@ -5,6 +5,6 @@ #include "flash_fwd_launch_template.h" template<> -void run_mha_fwd_<cutlass::half_t, 256>(Flash_fwd_params ¶ms, cudaStream_t stream) { - run_mha_fwd_hdim256<cutlass::half_t>(params, stream); +void run_mha_fwd_<cutlass::half_t, 256, false>(Flash_fwd_params ¶ms, cudaStream_t stream) { + run_mha_fwd_hdim256<cutlass::half_t, false>(params, stream); } |