ipex-llm/python/llm/test/inference_gpu
Ovo233 0dbce53464 LLM: Add decoder/layernorm unit tests (#10211)
* add decoder/layernorm unit tests

* update tests

* delete decoder tests

* address comments

* remove none type check

* restore nonetype checks

* delete nonetype checks; add decoder tests for Llama

* add gc

* deal with tuple output
2024-03-13 19:41:47 +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 [LLM] Recover arc ut test for Falcon (#10385) 2024-03-13 13:31:35 +08:00
test_transformers_api_attention.py [LLM] Recover arc ut test for Falcon (#10385) 2024-03-13 13:31:35 +08:00
test_transformers_api_final_logits.py [LLM] Recover arc ut test for Falcon (#10385) 2024-03-13 13:31:35 +08:00
test_transformers_api_layernorm.py LLM: Add decoder/layernorm unit tests (#10211) 2024-03-13 19:41:47 +08:00
test_transformers_api_mlp.py LLM: Add decoder/layernorm unit tests (#10211) 2024-03-13 19:41:47 +08:00
test_transformers_api_RMSNorm.py fix arc rms norm UT (#10394) 2024-03-13 13:09:15 +08:00