1234567891011 |
- // Copyright (c) 2023, Tri Dao.
- // Splitting the different head dimensions to different files to speed up
- // compilation. This file is auto-generated. See "generate_kernels.py"
- #include "flash_fwd_launch_template.h"
- template <>
- void run_mha_fwd_<cutlass::half_t, 192, true>(Flash_fwd_params& params,
- cudaStream_t stream) {
- run_mha_fwd_hdim192<cutlass::half_t, true>(params, stream);
- }
|