flash_fwd_mla_fp8_gfx938.cu 198 Bytes
Newer Older
zhanghj2's avatar
zhanghj2 committed
1
2
3
#include "flash_fwd_mla_kernel_fp8.h"

template void run_mha_fwd_splitkv_mla_fp8<cutlass::float_e4m3_t,cutlass::bfloat16_t,576>(Flash_fwd_mla_params &params, cudaStream_t stream, bool is_with_cat);