Antoni Baum 426ec4ec67 [1/n] Triton sampling kernel (#3186) ...

Co-authored-by: Roger Wang <136131678+ywang96@users.noreply.github.com>

2024-03-20 14:45:08 -07:00

..

allclose_default.py

[ROCm] Fix some kernels failed unit tests (#2498)

2024-02-05 14:25:36 -08:00

conftest.py

Support FP8-E5M2 KV Cache (#2279)

2024-01-28 16:43:54 -08:00

test_activation.py

Add kernel for GeGLU with approximate GELU (#3337)

2024-03-12 22:06:17 -07:00

test_attention.py

[ROCm] Fix some kernels failed unit tests (#2498)

2024-02-05 14:25:36 -08:00

test_cache.py

[Minor] More fix of test_cache.py CI test failure (#2750)

2024-02-06 11:38:38 -08:00

test_layernorm.py

Remove hardcoded device="cuda" to support more devices (#2503)

2024-02-01 15:46:39 -08:00

test_moe.py

Re-enable the 80 char line width limit (#3305)

2024-03-10 19:49:14 -07:00

test_pos_encoding.py

Add batched RoPE kernel (#3095)

2024-03-13 13:45:26 -07:00

test_prefix_prefill.py

fix lint

2024-03-15 22:23:38 -07:00

test_rand.py

[1/n] Triton sampling kernel (#3186)

2024-03-20 14:45:08 -07:00

test_sampler.py

[1/n] Triton sampling kernel (#3186)

2024-03-20 14:45:08 -07:00