| .. |
|
layers
|
0e8c46ae08
Run isort and black on test files
|
il y a 1 an |
|
losses
|
c7f32a8409
[CrossEntropy] Support precomputed LSE
|
il y a 3 mois |
|
models
|
73df3be7d5
Add test for BTLM init
|
il y a 11 mois |
|
modules
|
0e8c46ae08
Run isort and black on test files
|
il y a 1 an |
|
ops
|
f5b308e258
[LayerNorm] Rename layernorm.py -> layer_norm.py
|
il y a 11 mois |
|
pyproject.toml
|
73bd3f3bbb
Move pyproject.toml to flash-attn and tests dir to avoid PEP 517
|
il y a 1 an |
|
test_flash_attn.py
|
cdbbe844b1
minor changes to unpad_input test util func
|
il y a 2 mois |
|
test_flash_attn_ck.py
|
e2182cc21d
Support page kvcache in AMD ROCm (#1198)
|
il y a 2 mois |
|
test_flash_attn_triton_amd.py
|
b518517cb8
[AMD] Triton Backend for ROCm (#1203)
|
il y a 1 semaine |
|
test_rotary.py
|
cdbbe844b1
minor changes to unpad_input test util func
|
il y a 2 mois |
|
test_util.py
|
8cbc8a042f
small fixes
|
il y a 2 mois |
Michael Melesse
b518517cb8
[AMD] Triton Backend for ROCm (#1203)