squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
080438477f
vllm/tests/kernels
History
Zhuohan Li db09d4ad83
[FIX] Fix Alibi implementation in PagedAttention kernel (#945) 
* [FIX] Fix Alibi implementation in PagedAttention kernel

* Fix test_attention

* Fix

---------

Co-authored-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
Co-authored-by: Oliver-ss <yuansongwx@outlook.com>
2023-09-07 15:53:14 -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 [BugFix] Implement RoPE for GPT-J (#941) 2023-09-06 11:54:33 +09:00