Skip to content

Commit d95d03f

Browse files
authored
Parallel tests execution for tools, regression suites (#3962)
Before: * tools: 252.05s * regression: 624.95s * total: 1h15m After: * tools: 129.16s * regression: 40.32s * total: 1h10m
1 parent 321120e commit d95d03f

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

.github/workflows/build-test-reusable.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ jobs:
169169
170170
{
171171
echo SKIPLIST="$skiplist"
172-
echo TRITON_TEST_CMD="bash -v -x scripts/test-triton.sh --warning-reports --skip-pytorch-install --reports-dir $GITHUB_WORKSPACE/reports ${{ inputs.ignore_errors && '--ignore-errors' || '' }} $skiplist"
172+
echo TRITON_TEST_CMD="bash -x scripts/test-triton.sh --warning-reports --skip-pytorch-install --reports-dir $GITHUB_WORKSPACE/reports ${{ inputs.ignore_errors && '--ignore-errors' || '' }} $skiplist"
173173
} | tee -a $GITHUB_ENV
174174
175175
- name: Run Proton tests

scripts/test-triton.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -221,7 +221,7 @@ run_core_tests() {
221221
run_pytest_command -k "not test_line_info_interpreter" --verbose --device xpu language/test_line_info.py
222222

223223
TRITON_DISABLE_LINE_INFO=1 TRITON_TEST_SUITE=tools \
224-
run_pytest_command -k "not test_disam_cubin" --verbose tools
224+
run_pytest_command -n ${PYTEST_MAX_PROCESSES:-8} -k "not test_disam_cubin" --verbose tools
225225

226226
TRITON_DISABLE_LINE_INFO=1 TRITON_TEST_SUITE=intel \
227227
run_pytest_command -vvv -n ${PYTEST_MAX_PROCESSES:-8} --device xpu intel/
@@ -238,7 +238,7 @@ run_regression_tests() {
238238
cd $TRITON_PROJ/python/test/regression
239239

240240
TRITON_DISABLE_LINE_INFO=1 TRITON_TEST_SUITE=regression \
241-
run_pytest_command -vvv -s --device xpu . --ignore=test_performance.py
241+
run_pytest_command -vvv -n ${PYTEST_MAX_PROCESSES:-8} -s --device xpu . --ignore=test_performance.py
242242
}
243243

244244
run_interpreter_tests() {

0 commit comments

Comments
 (0)