| .. |
|
fmha
|
Bug fix: wrong smem_o write pointer for d=16
|
2022-06-25 15:18:33 -07:00 |
|
.DS_Store
|
Rename, add benchmarking script
|
2022-05-26 13:57:38 -07:00 |
|
fmha_block_dgrad_fp16_kernel_loop.sm80.cu
|
Rename, add benchmarking script
|
2022-05-26 13:57:38 -07:00 |
|
fmha_block_dgrad_kernel_1xN_loop.h
|
Refactor Gmem code to store q, k, v pointers separately
|
2022-06-12 16:37:32 -07:00 |
|
fmha_block_fprop_fp16_kernel.sm80.cu
|
Rename, add benchmarking script
|
2022-05-26 13:57:38 -07:00 |
|
fmha_block_fprop_kernel_1xN.h
|
Refactor Gmem code to store q, k, v pointers separately
|
2022-06-12 16:37:32 -07:00 |
|
fmha_blockmask.h
|
Rename, add benchmarking script
|
2022-05-26 13:57:38 -07:00 |
|
fmha_dgrad_fp16_kernel_loop.sm80.cu
|
Implement bwd for head dim 128
|
2022-06-11 17:52:36 -07:00 |
|
fmha_dgrad_kernel_1xN_loop.h
|
Fix race condition in backward pass (smem_dq)
|
2022-06-25 18:02:30 -07:00 |
|
fmha_fprop_fp16_kernel.sm80.cu
|
Implement bwd for head dim 128
|
2022-06-11 17:52:36 -07:00 |
|
fmha_fprop_kernel_1xN.h
|
Refactor Gmem code to store q, k, v pointers separately
|
2022-06-12 16:37:32 -07:00 |
|
fmha_kernel.h
|
Rename, add benchmarking script
|
2022-05-26 13:57:38 -07:00 |
|
fmha_utils.h
|
Rename, add benchmarking script
|
2022-05-26 13:57:38 -07:00 |
|
fmha.h
|
Refactor Gmem code to store q, k, v pointers separately
|
2022-06-12 16:37:32 -07:00 |
|
philox.cuh
|
Rename, add benchmarking script
|
2022-05-26 13:57:38 -07:00 |
