vllm/tests/kernels
2024-03-27 13:39:26 -07:00
..
allclose_default.py [ROCm] Fix some kernels failed unit tests (#2498) 2024-02-05 14:25:36 -08:00
conftest.py [CI] Try introducing isort. (#3495) 2024-03-25 07:59:47 -07:00
test_activation.py [CI] Try introducing isort. (#3495) 2024-03-25 07:59:47 -07:00
test_attention.py [CI] Try introducing isort. (#3495) 2024-03-25 07:59:47 -07:00
test_cache.py [CI] Try introducing isort. (#3495) 2024-03-25 07:59:47 -07:00
test_layernorm.py Remove hardcoded device="cuda" to support more devices (#2503) 2024-02-01 15:46:39 -08:00
test_moe.py [CI] Try introducing isort. (#3495) 2024-03-25 07:59:47 -07:00
test_pos_encoding.py [CI] Try introducing isort. (#3495) 2024-03-25 07:59:47 -07:00
test_prefix_prefill.py feat(benchmarks): Add Prefix Caching Benchmark to Serving Benchmark (#3277) 2024-03-27 13:39:26 -07:00
test_rand.py [CI] Try introducing isort. (#3495) 2024-03-25 07:59:47 -07:00
test_sampler.py [CI] Try introducing isort. (#3495) 2024-03-25 07:59:47 -07:00