We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 856f82f commit 73ddda4Copy full SHA for 73ddda4
scripts/test-triton.sh
@@ -581,8 +581,16 @@ run_triton_kernels_tests() {
581
echo "***************************************************"
582
cd $TRITON_PROJ/python/triton_kernels/tests
583
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
+
592
TRITON_TEST_SUITE=triton_kernels \
- run_pytest_command -vvv -n ${PYTEST_MAX_PROCESSES:-4} --device xpu .
593
+ run_pytest_command -vvv -n $max_procs --device xpu .
594
}
595
596
test_triton() {
0 commit comments