[CI] Fix changing XLA commit

Author: giordano

.github/workflows/test-gb-25.yml

@@ -99,18 +99,29 @@ jobs:
       - name: Load Julia packages from cache
         id: julia-cache
         uses: julia-actions/cache@v2
-      - name: Build local libReactant
-        timeout-minutes: 60
+      - name: Modify Enzyme-JAX commit
+        timeout-minutes: 1
         run: |
-          # Modify commit Enzyme-JAX commit
           sed -i.bak 's/ENZYMEXLA_COMMIT = ".*"/ENZYMEXLA_COMMIT = "${{ github.sha }}"/' ReactantExtra/WORKSPACE
 
+          cat ReactantExtra/WORKSPACE
+        working-directory: ${{ env.REACTANT_DIR }}/deps
+      - name: Modify XLA commit
+        timeout-minutes: 1
+        shell: bash
+        run: |
           # Modify XLA commit
           if [[ -n "${{ matrix.xla_commit }}" ]]; then
-              sed -i.bak \
-                  -e 's/xla_workspace(NEW_XLA_PATCHES)/xla_workspace(NEW_XLA_PATCHES, "${{ matrix.xla_commit }}") ReactantExtra/WORKSPACE
-          fi
+              sed -E -i.bak \
+                  -e 's/xla_workspace\(NEW_XLA_PATCHES(, .*)?\)/xla_workspace(NEW_XLA_PATCHES, "${{ matrix.xla_commit }}")/' \
+                  ReactantExtra/WORKSPACE
 
+              cat ReactantExtra/WORKSPACE
+          fi
+        working-directory: ${{ env.REACTANT_DIR }}/deps
+      - name: Build local libReactant
+        timeout-minutes: 60
+        run: |
           julia --project --color=yes -e 'using Pkg; Pkg.instantiate()'
           julia --project --color=yes -O0 build_local.jl --cc=clang --gcc_host_compiler_path=
         working-directory: ${{ env.REACTANT_DIR }}/deps