Add some flags for debugging pytest

Author: shunping

sdks/python/scripts/run_pytest.sh

@@ -129,15 +129,19 @@ if [[ -n "$test_paths" ]]; then
 fi
 pytest_command_args="$options $pyargs_section"
 
+export TRITON_CPU_ONLY=1
+export TORCH_COMPILE_DISABLE=1
+export PYTHONFAULTHANDLER=1
+
 # Run tests in parallel.
 echo "Running parallel tests with: pytest -m \"$marker_for_parallel_tests\" $pytest_command_args"
-pytest -v -rs -o junit_suite_name=${envname} \
+pytest -v -rs -n 1 -o junit_suite_name=${envname} \
   --junitxml=pytest_${envname}.xml -m "$marker_for_parallel_tests" -n 6 --import-mode=importlib ${pytest_args} ${pytest_command_args}
 status1=$?
 
 # Run tests sequentially.
 echo "Running sequential tests with: pytest -m \"$marker_for_sequential_tests\" $pytest_command_args"
-pytest -v -rs -o junit_suite_name=${envname}_no_xdist \
+pytest -v -rs -n 1 -o junit_suite_name=${envname}_no_xdist \
   --junitxml=pytest_${envname}_no_xdist.xml -m "$marker_for_sequential_tests" --import-mode=importlib ${pytest_args} ${pytest_command_args}
 status2=$?