Skip to content

Commit 73ddda4

Browse files
committed
update default value of max processes for B580
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 856f82f commit 73ddda4

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

scripts/test-triton.sh

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -581,8 +581,16 @@ run_triton_kernels_tests() {
581581
echo "***************************************************"
582582
cd $TRITON_PROJ/python/triton_kernels/tests
583583

584+
# available after `capture_runtime_env` call
585+
gpu_file="$TRITON_TEST_REPORTS_DIR/gpu.txt"
586+
if [[ -f "$gpu_file" ]] && grep -q "B580" "$gpu_file"; then
587+
max_procs=${PYTEST_MAX_PROCESSES:-1}
588+
else
589+
max_procs=${PYTEST_MAX_PROCESSES:-4}
590+
fi
591+
584592
TRITON_TEST_SUITE=triton_kernels \
585-
run_pytest_command -vvv -n ${PYTEST_MAX_PROCESSES:-4} --device xpu .
593+
run_pytest_command -vvv -n $max_procs --device xpu .
586594
}
587595

588596
test_triton() {

0 commit comments

Comments
 (0)