[test] add npu test yaml and add ascend a3 docker file (#9547)

Co-authored-by: jiaqiw09 <jiaqiw960714@gmail.com>
2025-12-18 12:50:38 +08:00 · 2025-11-30 09:37:08 +08:00
parent 22be45c78c
commit e43a972b25
33 changed files with 322 additions and 21 deletions
--- a/tests/model/model_utils/test_packing.py
+++ b/tests/model/model_utils/test_packing.py
@@ -18,6 +18,7 @@ import torch
 from llamafactory.model.model_utils.packing import get_seqlens_in_batch, get_unpad_data


+@pytest.mark.runs_on(["cpu","npu"])
@pytest.mark.parametrize(
    "attention_mask,golden_seq_lens",
    [