cutlass/examples/41_multi_head_attention
ANIKET SHIVAM b72cbf957d
CUTLASS 2.10 (#615)
Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
2022-09-03 18:48:46 -04:00
..
CMakeLists.txt CUTLASS 2.10 (#615) 2022-09-03 18:48:46 -04:00
fused_multihead_attention.cu CUTLASS 2.10 (#615) 2022-09-03 18:48:46 -04:00
gemm_attention.h CUTLASS 2.10 (#615) 2022-09-03 18:48:46 -04:00
gemm_grouped_with_softmax_visitor.h CUTLASS 2.10 (#615) 2022-09-03 18:48:46 -04:00