| .. |
|
epilogue
|
fMHA: Sync FW with xFormers (#828)
|
2023-02-22 23:25:31 -05:00 |
|
gemm
|
fMHA: Add backward pass (#844)
|
2023-04-06 20:44:58 -04:00 |
|
iterators
|
Fix for dangling references in the MHA example (#918)
|
2023-04-19 21:35:46 -04:00 |
|
transform
|
fMHA: Sync FW with xFormers (#828)
|
2023-02-22 23:25:31 -05:00 |
|
CMakeLists.txt
|
fMHA: Add backward pass (#844)
|
2023-04-06 20:44:58 -04:00 |
|
debug_utils.h
|
fMHA: Sync FW with xFormers (#828)
|
2023-02-22 23:25:31 -05:00 |
|
default_fmha_grouped.h
|
fMHA: Sync FW with xFormers (#828)
|
2023-02-22 23:25:31 -05:00 |
|
fmha_backward_test.py
|
fMHA: Add backward pass (#844)
|
2023-04-06 20:44:58 -04:00 |
|
fmha_grouped_problem_visitor.h
|
New updates for 2.11 (#775)
|
2023-01-20 16:32:57 -05:00 |
|
fmha_grouped.h
|
fMHA: Sync FW with xFormers (#828)
|
2023-02-22 23:25:31 -05:00 |
|
fused_multi_head_attention_backward.cu
|
fMHA: Add backward pass (#844)
|
2023-04-06 20:44:58 -04:00 |
|
fused_multihead_attention_fixed_seqlen.cu
|
Fix typos 2 (#842)
|
2023-03-09 23:22:56 -05:00 |
|
fused_multihead_attention_variable_seqlen.cu
|
Fix typos 2 (#842)
|
2023-03-09 23:22:56 -05:00 |
|
gemm_kernel_utils.h
|
fMHA: Add backward pass (#844)
|
2023-04-06 20:44:58 -04:00 |
|
kernel_backward.h
|
fMHA: Add backward pass (#844)
|
2023-04-06 20:44:58 -04:00 |
|
kernel_forward.h
|
fMHA: Add backward pass (#844)
|
2023-04-06 20:44:58 -04:00 |
|
piped_subprocess.py
|
fMHA: Add backward pass (#844)
|
2023-04-06 20:44:58 -04:00 |
