Skip to content

Commit 7bc952f

Browse files
giordanofelixwqp
authored andcommitted
Fix xla commit in GB-25 code
1 parent a36cc1b commit 7bc952f

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

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

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,17 @@ jobs:
121121
- name: Build local libReactant
122122
timeout-minutes: 60
123123
run: |
124+
# Modify commit Enzyme-JAX commit
125+
sed -i.bak 's/ENZYMEXLA_COMMIT = ".*"/ENZYMEXLA_COMMIT = "${{ github.sha }}"/' ReactantExtra/WORKSPACE
126+
127+
# Modify XLA commit
128+
if [[ -n "${{ matrix.xla_commit }}" ]]; then
129+
sed -E -i.bak \
130+
-e 's?load\("@jax//third_party/xla:revision.bzl", "XLA_COMMIT", "XLA_SHA256"\)??' \
131+
-e 's/(# )?XLA_COMMIT = ".*"/XLA_COMMIT = "${{ matrix.xla_commit }}"/' \
132+
-e 's/(# )?XLA_SHA256 = ""/XLA_SHA256 = ""/' \
133+
ReactantExtra/WORKSPACE
134+
fi
124135
julia --project --color=yes -e 'using Pkg; Pkg.instantiate()'
125136
julia --project --color=yes -O0 build_local.jl --cc=clang --gcc_host_compiler_path=
126137
working-directory: ${{ env.REACTANT_DIR }}/deps

0 commit comments

Comments
 (0)