ipex-llm/python/llm/test/inference_gpu
Ovo233 60e11b6739 LLM: Add mlp layer unit tests (#10200)
* add mlp layer unit tests

* add download baichuan-13b

* exclude llama for now

* install additional packages

* rename bash file

* switch to Baichuan2

* delete attention related code

* fix name errors in yml file
2024-02-22 13:44:45 +08:00
..
test_layer_fast_rope.py Support fast rope for training (#9745) 2024-01-17 15:51:38 +08:00
test_transformers_api.py Unit test on final logits and the logits of the last attention layer (#10093) 2024-02-07 14:25:36 +08:00
test_transformers_api_attention.py Unit test on final logits and the logits of the last attention layer (#10093) 2024-02-07 14:25:36 +08:00
test_transformers_api_final_logits.py Unit test on final logits and the logits of the last attention layer (#10093) 2024-02-07 14:25:36 +08:00
test_transformers_api_mlp.py LLM: Add mlp layer unit tests (#10200) 2024-02-22 13:44:45 +08:00