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