ci: enable/disable gil as needed

Author: cpcloud

ci/tools/run-tests

@@ -26,8 +26,10 @@ FREE_THREADING=""
 if python -c 'import sys; assert not sys._is_gil_enabled()' 2> /dev/null; then
   FREE_THREADING+="-ft"
   PYTEST_ARGS+=("--require-gil-disabled" "--threads" "${FREE_THREADING_TEST_THREADS}" "--iterations" "${FREE_THREADING_TEST_ITERATIONS}")
+  PYTHON_GIL=0
 else
   PYTEST_ARGS+=("-p" "no:freethreaded")
+  PYTHON_GIL=1
 fi
 
 
@@ -46,7 +48,7 @@ if [[ "${test_module}" == "pathfinder" ]]; then
       "LD:${CUDA_PATHFINDER_TEST_LOAD_NVIDIA_DYNAMIC_LIB_STRICTNESS} " \
       "FH:${CUDA_PATHFINDER_TEST_FIND_NVIDIA_HEADERS_STRICTNESS}"
   pwd
-  pytest "${PYTEST_ARGS[@]}" -ra -s -v tests/ |& tee /tmp/pathfinder_test_log.txt
+  PYTHON_GIL="$PYTHON_GIL" pytest "${PYTEST_ARGS[@]}" -ra -s -v tests/ |& tee /tmp/pathfinder_test_log.txt
   # Fail if no "INFO test_" lines are found; capture line count otherwise
   line_count=$(grep '^INFO test_' /tmp/pathfinder_test_log.txt | wc -l)
   echo "Number of \"INFO test_\" lines: $line_count"
@@ -66,9 +68,9 @@ elif [[ "${test_module}" == "bindings" ]]; then
   pushd ./cuda_bindings
   echo "Running bindinds tests"
   pwd
-  ${SANITIZER_CMD} pytest "${PYTEST_ARGS[@]}" -rxXs -v tests/
+  PYTHON_GIL="$PYTHON_GIL" ${SANITIZER_CMD} pytest "${PYTEST_ARGS[@]}" -rxXs -v tests/
   if [[ "${SKIP_CYTHON_TEST}" == 0 ]]; then
-    ${SANITIZER_CMD} pytest "${PYTEST_ARGS[@]}" -rxXs -v tests/cython
+    PYTHON_GIL="$PYTHON_GIL" ${SANITIZER_CMD} pytest "${PYTEST_ARGS[@]}" -rxXs -v tests/cython
   fi
   popd
 elif [[ "${test_module}" == "core" ]]; then
@@ -103,11 +105,11 @@ elif [[ "${test_module}" == "core" ]]; then
   pushd ./cuda_core
   echo "Running core tests"
   pwd
-  ${SANITIZER_CMD} pytest "${PYTEST_ARGS[@]}" -rxXs -v tests/
+  PYTHON_GIL="$PYTHON_GIL" ${SANITIZER_CMD} pytest "${PYTEST_ARGS[@]}" -rxXs -v tests/
   # Currently our CI always installs the latest bindings (from either major version).
   # This is not compatible with the test requirements.
   if [[ "${SKIP_CYTHON_TEST}" == 0 ]]; then
-    ${SANITIZER_CMD} pytest "${PYTEST_ARGS[@]}" -rxXs -v tests/cython
+    PYTHON_GIL="$PYTHON_GIL" ${SANITIZER_CMD} pytest "${PYTEST_ARGS[@]}" -rxXs -v tests/cython
   fi
   popd
 fi