squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
757275f279
cutlass/examples/41_fused_multi_head_attention/iterators
History
dan_the_3rd 146d314057
Update fMHA kernels (#992) 
* Update fMHA kernels

Upstream recent changes to fMHA that we did in xFormers.
Previous version in CUTLASS: facebookresearch/xformers@b6be33a
Updating to: facebookresearch/xformers@55a4798

* minor changes

* make var work

---------

Co-authored-by: danthe3rd <danthe3rd>
Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
2023-07-12 22:30:46 -04:00
..
default_warp_iterator_from_smem.h Update fMHA kernels (#992) 2023-07-12 22:30:46 -04:00
epilogue_predicated_tile_iterator.h xFormer updates to fMHA FW (#773) 2023-02-08 23:00:10 -05:00
make_residual_last.h New updates for 2.11 (#775) 2023-01-20 16:32:57 -05:00
predicated_tile_access_iterator_residual_last.h Fix for dangling references in the MHA example (#918) 2023-04-19 21:35:46 -04:00
predicated_tile_iterator_residual_last.h New updates for 2.11 (#775) 2023-01-20 16:32:57 -05:00
transpose_warp_iterator.h Update fMHA kernels (#992) 2023-07-12 22:30:46 -04:00
warp_iterator_from_smem.h Update fMHA kernels (#992) 2023-07-12 22:30:46 -04:00