vllm/tests/kernels
Woosuk Kwon e67b4f2c2a
Use FP32 in RoPE initialization (#1004)
Co-authored-by: One <imone@tuta.io>
2023-09-11 00:26:35 -07:00
..
conftest.py Clean up kernel unit tests (#938) 2023-09-05 16:57:38 -07:00
test_activation.py Clean up kernel unit tests (#938) 2023-09-05 16:57:38 -07:00
test_attention.py [FIX] Fix Alibi implementation in PagedAttention kernel (#945) 2023-09-07 15:53:14 -07:00
test_cache.py Clean up kernel unit tests (#938) 2023-09-05 16:57:38 -07:00
test_layernorm.py Clean up kernel unit tests (#938) 2023-09-05 16:57:38 -07:00
test_pos_encoding.py Use FP32 in RoPE initialization (#1004) 2023-09-11 00:26:35 -07:00