| .. |
|
cutlass @ 756c351b49
|
74b0761ff7
[FA3] BF16 forward
|
5 ヶ月 前 |
|
flash_attn
|
40e534a7f6
Implement cache_leftpad
|
5 ヶ月 前 |
|
ft_attention
|
50896ec574
Make nvcc threads configurable via environment variable (#885)
|
9 ヶ月 前 |
|
fused_dense_lib
|
50896ec574
Make nvcc threads configurable via environment variable (#885)
|
9 ヶ月 前 |
|
fused_softmax
|
50896ec574
Make nvcc threads configurable via environment variable (#885)
|
9 ヶ月 前 |
|
layer_norm
|
50896ec574
Make nvcc threads configurable via environment variable (#885)
|
9 ヶ月 前 |
|
rotary
|
50896ec574
Make nvcc threads configurable via environment variable (#885)
|
9 ヶ月 前 |
|
xentropy
|
50896ec574
Make nvcc threads configurable via environment variable (#885)
|
9 ヶ月 前 |
Tri Dao
74b0761ff7
[FA3] BF16 forward