Kunshang Ji 96b6f475dd Remove hardcoded device="cuda" to support more devices (#2503) ... Co-authored-by: Jiang Li <jiang1.li@intel.com> Co-authored-by: Kunshang Ji <kunshang.ji@intel.com>		2024-02-01 15:46:39 -08:00
..
conftest.py	Support FP8-E5M2 KV Cache (#2279)	2024-01-28 16:43:54 -08:00
test_activation.py	Remove hardcoded device="cuda" to support more devices (#2503)	2024-02-01 15:46:39 -08:00
test_attention.py	Remove hardcoded device="cuda" to support more devices (#2503)	2024-02-01 15:46:39 -08:00
test_cache.py	Remove hardcoded device="cuda" to support more devices (#2503)	2024-02-01 15:46:39 -08:00
test_layernorm.py	Remove hardcoded device="cuda" to support more devices (#2503)	2024-02-01 15:46:39 -08:00
test_moe.py	Add unit test for Mixtral MoE layer (#2677)	2024-01-31 14:34:17 -08:00
test_pos_encoding.py	Remove hardcoded device="cuda" to support more devices (#2503)	2024-02-01 15:46:39 -08:00
test_prefix_prefill.py	Remove hardcoded device="cuda" to support more devices (#2503)	2024-02-01 15:46:39 -08:00