Skip to content

Commit 09a0c8f

Browse files
WIP: Set n_workers for vgf_fp and tosa_fp tests
Signed-off-by: Oscar Andersson <oscar.andersson@arm.com> Change-Id: I3c0d5f6abb9a21373baa5b3e101286af1bdb053c
1 parent 8ad3111 commit 09a0c8f

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

.ci/scripts/test_backend.sh

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#!/usr/bin/env bash
22
# Copyright (c) Meta Platforms, Inc. and affiliates.
33
# All rights reserved.
4-
# Copyright 2025 Arm Limited and/or its affiliates.
4+
# Copyright 2025-2026 Arm Limited and/or its affiliates.
55
#
66
# This source code is licensed under the BSD-style license found in the
77
# LICENSE file in the root directory of this source tree.
@@ -94,7 +94,12 @@ CMAKE_ARGS="$EXTRA_BUILD_ARGS" ${CONDA_RUN_CMD} $SETUP_SCRIPT --build-tool cmake
9494
GOLDEN_DIR="${ARTIFACT_DIR}/golden-artifacts"
9595
export GOLDEN_ARTIFACTS_DIR="${GOLDEN_DIR}"
9696

97+
PYTEST_WORKERS="auto"
98+
if [[ "$SUITE" == "models" ]] && [[ "$FLOW" == "arm_tosa_fp" || "$FLOW" == "arm_vgf_fp" ]]; then
99+
PYTEST_WORKERS="1"
100+
fi
101+
97102
EXIT_CODE=0
98-
${CONDA_RUN_CMD} pytest -c /dev/null -n auto backends/test/suite/$SUITE/ -m flow_$FLOW --json-report --json-report-file="$REPORT_FILE" || EXIT_CODE=$?
103+
${CONDA_RUN_CMD} pytest -c /dev/null -n "$PYTEST_WORKERS" backends/test/suite/$SUITE/ -m flow_$FLOW --json-report --json-report-file="$REPORT_FILE" || EXIT_CODE=$?
99104
# Generate markdown summary.
100105
${CONDA_RUN_CMD} python -m executorch.backends.test.suite.generate_markdown_summary_json "$REPORT_FILE" > ${GITHUB_STEP_SUMMARY:-"step_summary.md"} --exit-code $EXIT_CODE

0 commit comments

Comments
 (0)