Repository navigation
Commit d46e620
authored
Arm backend: Add serialised xlarge VKML model suite (pytorch#21518)
The DeepSeek-R1-Distill-Qwen layer tests were re-landed after the TOSA
model shard OOM was addressed by a serialised xlarge TOSA suite.
CommitlyExtended still showed memory pressure in the VKML model shard.
The normal VKML model shard uses pytest-xdist auto parallelism, so
xlarge VGF exports can also be resident at the same time.
Add a serialised xlarge VKML model suite while keeping the normal VKML
model shard parallelised. Mark the DeepSeek VGF layer tests as xlarge so
they are excluded from normal VKML and can be routed through the
serialised suite.
cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils
@Sebastian-Larsson @robell @rascani1 parent 9789eea commit d46e620
4 files changed
Lines changed: 479 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 | | |
| |||
| 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