| .. |
|
epilogue
|
Updates for 3.4 release. (#1305)
|
2024-01-16 13:42:51 -05:00 |
|
gemm
|
Updates for 3.4 release. (#1305)
|
2024-01-16 13:42:51 -05:00 |
|
iterators
|
Updates for 3.4 release. (#1305)
|
2024-01-16 13:42:51 -05:00 |
|
transform
|
Updates for 3.4 release. (#1305)
|
2024-01-16 13:42:51 -05:00 |
|
CMakeLists.txt
|
fMHA: Add backward pass (#844)
|
2023-04-06 20:44:58 -04:00 |
|
debug_utils.h
|
Updates for 3.4 release. (#1305)
|
2024-01-16 13:42:51 -05:00 |
|
default_fmha_grouped.h
|
Update fMHA kernels (#992)
|
2023-07-12 22:30:46 -04:00 |
|
fmha_backward_test.py
|
Updates for 3.4 release. (#1305)
|
2024-01-16 13:42:51 -05:00 |
|
fmha_grouped_problem_visitor.h
|
New updates for 2.11 (#775)
|
2023-01-20 16:32:57 -05:00 |
|
fmha_grouped.h
|
Update fMHA kernels (#992)
|
2023-07-12 22:30:46 -04:00 |
|
fused_multi_head_attention_backward.cu
|
Updates for 3.4 release. (#1305)
|
2024-01-16 13:42:51 -05:00 |
|
fused_multihead_attention_fixed_seqlen.cu
|
Updates for 3.4 release. (#1305)
|
2024-01-16 13:42:51 -05:00 |
|
fused_multihead_attention_variable_seqlen.cu
|
Updates for 3.4 release. (#1305)
|
2024-01-16 13:42:51 -05:00 |
|
gemm_kernel_utils.h
|
Updates for 3.4 release. (#1305)
|
2024-01-16 13:42:51 -05:00 |
|
kernel_backward.h
|
Updates for 3.4 release. (#1305)
|
2024-01-16 13:42:51 -05:00 |
|
kernel_forward.h
|
Updates for 3.4 release. (#1305)
|
2024-01-16 13:42:51 -05:00 |
|
piped_subprocess.py
|
Updates for 3.4 release. (#1305)
|
2024-01-16 13:42:51 -05:00 |
