// Copyright (c) 2024, Jay Shah, Ganesh Bikshandi, Ying Zhang, Vijay Thakkar, Pradeep Ramani, Tri Dao. | |
// Splitting the different template instantiations to different files to speed up compilation. | |
// This file is auto-generated. See "generate_kernels.py" | |
template<> | |
void run_mha_bwd_<80, cutlass::half_t, 192, false>(Flash_bwd_params ¶ms, cudaStream_t stream) { | |
run_mha_bwd_hdim192<80, cutlass::half_t, false>(params, stream); | |
} | |
template<> | |
void run_mha_bwd_<86, cutlass::half_t, 192, false>(Flash_bwd_params ¶ms, cudaStream_t stream) { | |
run_mha_bwd_hdim192<86, cutlass::half_t, false>(params, stream); | |
} | |