add concurrency settings and remove condition that disables running on forks

Author: nitins17

.github/workflows/bazel_cpu_rbe.yml

@@ -1,9 +1,9 @@
 name: CI - Bazel CPU tests (RBE)
 
 on:
-  # pull_request:
-  #   branches:
-  #     - main
+  pull_request:
+    branches:
+      - main
   workflow_dispatch:
     inputs:
       halt-for-connection:
@@ -21,10 +21,13 @@ concurrency:
 
 jobs:
   run_tests:
-    if: github.event.repository.fork == false
+    defaults:
+      run:
+        # Explicitly set the shell to bash to override the default Windows environment, i.e, cmd.
+        shell: bash
     strategy:
       matrix:
-        runner: ["windows-x86-n2-64"] #, "linux-x86-n2-16", "linux-arm64-t2a-16"]
+        runner: ["windows-x86-n2-64", "linux-x86-n2-16", "linux-arm64-t2a-16"]
 
     runs-on: ${{ matrix.runner }}
     container: ${{ (contains(matrix.runner, 'linux-x86') && 'us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest') ||

.github/workflows/bazel_gpu_non_rbe.yml

@@ -15,6 +15,10 @@ on:
         - 'yes'
         - 'no'
 
+concurrency:
+  group: ${{ github.workflow }}-${{ github.head_ref || github.ref }}
+  cancel-in-progress: true
+
 jobs:
   build_artifacts:
     name: "Build the jaxlib and CUDA plugins using latest XLA"

.github/workflows/bazel_gpu_rbe.yml

@@ -1,9 +1,9 @@
 name: CI - Bazel GPU tests (RBE)
 
 on:
-  # pull_request:
-  #   branches:
-  #     - main
+  pull_request:
+    branches:
+      - main
   workflow_dispatch:
     inputs:
       halt-for-connection:
@@ -21,7 +21,6 @@ concurrency:
 
 jobs:
   run_tests:
-    if: github.event.repository.fork == false
     strategy:
       matrix:
         runner: ["linux-x86-n2-16"]

.github/workflows/pytest_cpu.yml

@@ -14,6 +14,10 @@ on:
         options:
         - 'yes'
 
+concurrency:
+  group: ${{ github.workflow }}-${{ github.head_ref || github.ref }}
+  cancel-in-progress: true
+
 jobs:
   build_jaxlib_artifact:
     name: "Build the jaxlib aritfact using latest XLA"

.github/workflows/pytest_gpu.yml

@@ -15,6 +15,10 @@ on:
         - 'yes'
         - 'no'
 
+concurrency:
+  group: ${{ github.workflow }}-${{ github.head_ref || github.ref }}
+  cancel-in-progress: true
+
 jobs:
   build_artifacts:
     name: "Build the jaxlib and CUDA plugins using latest XLA"

.github/workflows/pytest_tpu.yml

@@ -15,6 +15,10 @@ on:
         - 'yes'
         - 'no'
 
+concurrency:
+  group: ${{ github.workflow }}-${{ github.head_ref || github.ref }}
+  cancel-in-progress: true
+
 jobs:
   run_tpu_tests:
     strategy: