Skip to content

Commit b768f79

Browse files
authored
Change pytest-xdist distribution algorithm to worksteal (#4028)
This is similar to the `load` method (default), but `worksteal` should handle tests with significantly differing duration better. Core time: 41m -> 25m Total time: 53m -> 36m Compared for the following runs: * Before: https://github.com/intel/intel-xpu-backend-for-triton/actions/runs/14684415545 * After: https://github.com/intel/intel-xpu-backend-for-triton/actions/runs/14684687453 Signed-off-by: Pavel Chekin <[email protected]>
1 parent 95d0f06 commit b768f79

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

scripts/pytest-utils.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,9 @@ fi
1616
TRITON_TEST_SKIPLIST_DIR="$(cd "$TRITON_TEST_SKIPLIST_DIR" && pwd)"
1717

1818
pytest() {
19-
pytest_extra_args=()
19+
pytest_extra_args=(
20+
"--dist=worksteal"
21+
)
2022

2123
if [[ -v TRITON_TEST_SUITE && $TRITON_TEST_REPORTS = true ]]; then
2224
mkdir -p "$TRITON_TEST_REPORTS_DIR"

0 commit comments

Comments
 (0)