Repository navigation
Commit e798c4d
authored
Arm backend: Add serialized xlarge TOSA model suite (pytorch#21492)
The DeepSeek-R1-Distill-Qwen layer-test PRs were reverted because they
caused OOMs in the no-driver Arm TOSA model pytest job.
The normal TOSA model shard should keep using pytest-xdist auto
parallelism so unrelated model tests do not slow down. Add a separate
serialized suite for xlarge TOSA model tests instead.
The reland of the DeepSeek layer tests can mark the memory-heavy cases
as xlarge and route them through this suite. This also gives internal CI
an explicit option for running any xlarge TOSA model coverage serially.
cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils
@Sebastian-Larsson @robell @rascani
Signed-off-by: Baris Demir <baris.demir@arm.com>1 parent b26b9ac commit e798c4d
1 file changed
Lines changed: 12 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
115 | 115 | | |
116 | 116 | | |
117 | 117 | | |
| 118 | + | |
| 119 | + | |
| 120 | + | |
| 121 | + | |
| 122 | + | |
| 123 | + | |
| 124 | + | |
| 125 | + | |
| 126 | + | |
| 127 | + | |
| 128 | + | |
| 129 | + | |
118 | 130 | | |
119 | 131 | | |
120 | 132 | | |
| |||
0 commit comments