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