Set timeout for artifact building and "run tests" steps

Also, use a conditional expression in the continuous workflow to control concurrent runs. We don't want to cancel runs on multiple pushes to main or release branch.

PiperOrigin-RevId: 716780290

.github/workflows/build_artifacts.yml

@@ -128,6 +128,7 @@ jobs:
         with:
           halt-dispatch-input: ${{ inputs.halt-for-connection }}
       - name: Build ${{ inputs.artifact }}
+        timeout-minutes: 30
         run: ./ci/build_artifacts.sh "${{ inputs.artifact }}"
       - name: Upload artifacts to a GCS bucket (non-Windows runs)
         if: >-

.github/workflows/pytest_cpu.yml

@@ -97,4 +97,5 @@ jobs:
         with:
           halt-dispatch-input: ${{ inputs.halt-for-connection }}
       - name: Run Pytest CPU tests
+        timeout-minutes: 60
         run: ./ci/run_pytest_cpu.sh

.github/workflows/pytest_cuda.yml

@@ -86,4 +86,5 @@ jobs:
         with:
           halt-dispatch-input: ${{ inputs.halt-for-connection }}
       - name: Run Pytest CUDA tests
+        timeout-minutes: 60
         run: ./ci/run_pytest_cuda.sh

.github/workflows/wheel_tests_continuous.yml

@@ -19,7 +19,7 @@ on:
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.head_ref || github.ref }}
-  cancel-in-progress: true
+  cancel-in-progress: ${{ !contains(github.ref, 'release/') && github.ref != 'main' }}
 
 jobs:
   build-jaxlib-artifact: