vllm/tests/kernels
2024-04-18 00:51:28 -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 [Misc] Add indirection layer for custom ops (#3913) 2024-04-10 20:26:07 -07:00
test_cache.py [Misc] Add indirection layer for custom ops (#3913) 2024-04-10 20:26:07 -07:00
test_layernorm.py [Kernel] Layernorm performance optimization (#3662) 2024-03-30 14:26:38 -07:00
test_moe.py [Core] Set linear_weights directly on the layer (#3977) 2024-04-11 16:35:51 -04:00
test_pos_encoding.py [CI] Try introducing isort. (#3495) 2024-03-25 07:59:47 -07:00
test_prefix_prefill.py [Bugfix][Kernel] allow non-power-of-two head sizes in prefix prefill (#4128) 2024-04-18 00:51:28 -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