Skip to content

Commit e798c4d

Browse files
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

File tree

‎backends/arm/test/test_arm_backend.sh‎

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,18 @@ test_pytest_models_tosa() {
115115
echo "${TEST_SUITE_NAME}: PASS"
116116
}
117117

118+
test_pytest_models_tosa_xlarge() {
119+
echo "${TEST_SUITE_NAME}: Run xlarge pytest models for TOSA"
120+
121+
# Install model dependencies for pytest
122+
source backends/arm/scripts/install_models_for_test.sh
123+
124+
pytest "${PYTEST_RETRY_ARGS[@]}" --verbose --color=yes \
125+
--numprocesses=1 --durations=0 \
126+
backends/arm/test/models -k tosa -m xlarge
127+
echo "${TEST_SUITE_NAME}: PASS"
128+
}
129+
118130
test_run_tosa() {
119131
echo "${TEST_SUITE_NAME}: Test TOSA delegate examples with run.sh"
120132

0 commit comments

Comments
 (0)