.. |
composable_kernel @ 8182976c37
|
d8f104e97a
Support AMD ROCm on FlashAttention 2 (#1010)
|
4 bulan lalu |
cutlass @ 756c351b49
|
74b0761ff7
[FA3] BF16 forward
|
5 bulan lalu |
flash_attn
|
65f723bb9a
Split bwd into more .cu files to speed up compilation
|
4 bulan lalu |
flash_attn_ck
|
d8f104e97a
Support AMD ROCm on FlashAttention 2 (#1010)
|
4 bulan lalu |
ft_attention
|
50896ec574
Make nvcc threads configurable via environment variable (#885)
|
9 bulan lalu |
fused_dense_lib
|
50896ec574
Make nvcc threads configurable via environment variable (#885)
|
9 bulan lalu |
fused_softmax
|
50896ec574
Make nvcc threads configurable via environment variable (#885)
|
9 bulan lalu |
layer_norm
|
50896ec574
Make nvcc threads configurable via environment variable (#885)
|
9 bulan lalu |
rotary
|
50896ec574
Make nvcc threads configurable via environment variable (#885)
|
9 bulan lalu |
xentropy
|
50896ec574
Make nvcc threads configurable via environment variable (#885)
|
9 bulan lalu |