Commit c995192
authored
Arm backend: Add DeepSeek-R1-Distill-Qwen layer tests (#21010)
This patch adds Arm backend layer tests for
DeepSeek-R1-Distill-Qwen-1.5B. The tests use the checkpoint
configuration from the Hugging Face model and the upstream Qwen2 layer
implementations that back this distilled model.
The test structure follows the existing Qwen3-VL layer-test approach.
The config helper constructs the checkpoint-sized model configuration
directly, the test file uses layer-level wrappers with dataclass-driven
test cases, and coverage includes TOSA FP, TOSA BF16 reference-model,
VGF no-quant, and VGF BF16 no-quant runtime paths.
Token embedding is excluded because the full checkpoint embedding
allocation is too large for regular CI. VGF quant coverage is also left
out of this patch so CI resources stay focused on the requested BF16
path and export coverage.
The covered layers include rotary embedding, rotary application, KV
repetition, attention, RMSNorm, MLP, decoder layer, and final norm.
cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils
@Sebastian-Larsson @robell @rascani
---------
Signed-off-by: Baris Demir <baris.demir@arm.com>1 parent b532ec1 commit c995192
3 files changed
Lines changed: 461 additions & 0 deletions
File tree
- backends/arm
- test/models/DeepSeek_R1_Distill_Qwen
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
5 | 5 | | |
6 | 6 | | |
7 | 7 | | |
| 8 | + | |
8 | 9 | | |
9 | 10 | | |
10 | 11 | | |
| |||
Lines changed: 42 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
| 27 | + | |
| 28 | + | |
| 29 | + | |
| 30 | + | |
| 31 | + | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
| 41 | + | |
| 42 | + | |
0 commit comments