wbn dacaf5a400 Replace head_mapping params with num_kv_heads to attention kernel. (#1997) ... Co-authored-by: wangguoya <wangguoya@baidu.com> Co-authored-by: Yang Zhao <zhaoyangstar@foxmail.com>		2023-12-10 10:12:53 -08:00
..
conftest.py	Clean up kernel unit tests (#938)	2023-09-05 16:57:38 -07:00
test_activation.py	Add PyTorch-native implementation of custom layers (#1898)	2023-12-02 21:18:40 -08:00
test_attention.py	Replace head_mapping params with num_kv_heads to attention kernel. (#1997)	2023-12-10 10:12:53 -08:00
test_cache.py	[Build] Avoid building too many extensions (#1624)	2023-11-23 16:31:19 -08:00
test_layernorm.py	Add PyTorch-native implementation of custom layers (#1898)	2023-12-02 21:18:40 -08:00
test_pos_encoding.py	Add PyTorch-native implementation of custom layers (#1898)	2023-12-02 21:18:40 -08:00