cutlass/examples/41_fused_multi_head_attention
Vijay Thakkar 629f4653c3
CUTLASS 3.5.0 (#1411)
2024-03-19 17:51:04 -04:00
..
epilogue Update license year (#1306) 2024-01-16 14:37:22 -05:00
gemm CUTLASS 3.5.0 (#1411) 2024-03-19 17:51:04 -04:00
iterators Update license year (#1306) 2024-01-16 14:37:22 -05:00
transform Update license year (#1306) 2024-01-16 14:37:22 -05:00
CMakeLists.txt Update license year (#1306) 2024-01-16 14:37:22 -05:00
debug_utils.h CUTLASS 3.5.0 (#1411) 2024-03-19 17:51:04 -04:00
default_fmha_grouped.h Update license year (#1306) 2024-01-16 14:37:22 -05:00
fmha_backward_test.py Updates for 3.4 release. (#1305) 2024-01-16 13:42:51 -05:00
fmha_grouped.h CUTLASS 3.5.0 (#1411) 2024-03-19 17:51:04 -04:00
fmha_grouped_problem_visitor.h Update license year (#1306) 2024-01-16 14:37:22 -05:00
fused_multi_head_attention_backward.cu Update license year (#1306) 2024-01-16 14:37:22 -05:00
fused_multihead_attention_fixed_seqlen.cu CUTLASS 3.5.0 (#1411) 2024-03-19 17:51:04 -04:00
fused_multihead_attention_variable_seqlen.cu CUTLASS 3.5.0 (#1411) 2024-03-19 17:51:04 -04:00
gemm_kernel_utils.h Update license year (#1306) 2024-01-16 14:37:22 -05:00
kernel_backward.h CUTLASS 3.5.0 (#1411) 2024-03-19 17:51:04 -04:00
kernel_forward.h CUTLASS 3.5.0 (#1411) 2024-03-19 17:51:04 -04:00
piped_subprocess.py Updates for 3.4 release. (#1305) 2024-01-16 13:42:51 -05:00