@@ -56,23 +56,26 @@ jobs:
5656 apt-get update
5757 apt-get install -y binutils ninja-build cmake gcc g++ python3 python3-dev
5858
59- - uses : actions/checkout@v4
59+ - name : Check out Reactant
60+ uses : actions/checkout@v6
6061 with :
6162 path : ' Reactant'
6263
63-
64- - uses : actions/checkout@v4
64+ - name : Check out LLVM
65+ uses : actions/checkout@v6
6566 with :
6667 repository : ' llvm/llvm-project'
6768 ref : ' 4749bf56a65e38ee7b05ac7f9fe261aab6cb5bc6'
6869 path : ' llvm-project'
69-
70- - uses : actions/checkout@v4
70+
71+ - name : Check out Enzyme-JAX
72+ uses : actions/checkout@v6
7173 with :
7274 repository : ' EnzymeAD/Enzyme-JaX'
7375 path : ' Enzyme-JaX'
74-
75- - uses : actions/checkout@v4
76+
77+ - name : Check out Enzyme GPU Tests
78+ uses : actions/checkout@v6
7679 with :
7780 repository : ' wsmoses/Enzyme-GPU-Tests'
7881 path : ' Enzyme-GPU-Tests'
9497 working-directory : ' llvm-project'
9598 run : echo "sha_short=$(git rev-parse --short HEAD)" >> $GITHUB_OUTPUT
9699
97-
98100 - name : Cache MLIR
99101 id : cache-mlir
100102 uses : actions/cache@v4
@@ -129,31 +131,31 @@ jobs:
129131 # Share repository cache between workflows.
130132 repository-cache : true
131133 bazelisk-version : 1.x
132-
134+
133135 - name : Set BAZEL_FLAGS
134136 shell : bash
135137 run : |
136- set -e
137- BAZEL_FLAGS=""
138- BAZEL_FLAGS="${BAZEL_FLAGS} --linkopt=-fuse-ld=lld"
139- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=TPU_TOPOLOGY"
140- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=TPU_WORKER_ID"
141- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=TPU_SKIP_MDS_QUERY=true"
142- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=TPU_TOPOLOGY_WRAP"
143- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=TPU_CHIPS_PER_HOST_BOUNDS"
144- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=TPU_ACCELERATOR_TYPE"
145- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=TPU_RUNTIME_METRICS_PORTS"
146- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=TPU_TOPOLOGY_ALT"
147- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=TPU_HOST_BOUNDS"
148- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=TPU_WORKER_HOSTNAMES"
149- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=CHIPS_PER_HOST_BOUNDS"
150- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=HOST_BOUNDS"
151- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=ALT=false"
152- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=WRAP"
153- BAZEL_FLAGS="${BAZEL_FLAGS} --test_env=VBAR_CONTROL_SERVICE_URL"
154- BAZEL_FLAGS="${BAZEL_FLAGS} --repo_env=CC=`which clang`"
155- BAZEL_FLAGS="${BAZEL_FLAGS} --define=using_clang=true"
156- echo "BAZEL_FLAGS=${BAZEL_FLAGS}" >> "${GITHUB_ENV}"
138+ BAZEL_FLAGS=(
139+ --linkopt=-fuse-ld=lld
140+ --test_env=TPU_TOPOLOGY
141+ --test_env=TPU_WORKER_ID
142+ --test_env=TPU_SKIP_MDS_QUERY=true
143+ --test_env=TPU_TOPOLOGY_WRAP
144+ --test_env=TPU_CHIPS_PER_HOST_BOUNDS
145+ --test_env=TPU_ACCELERATOR_TYPE
146+ --test_env=TPU_RUNTIME_METRICS_PORTS
147+ --test_env=TPU_TOPOLOGY_ALT
148+ --test_env=TPU_HOST_BOUNDS
149+ --test_env=TPU_WORKER_HOSTNAMES
150+ --test_env=CHIPS_PER_HOST_BOUNDS
151+ --test_env=HOST_BOUNDS
152+ --test_env=ALT=false
153+ --test_env=WRAP
154+ --test_env=VBAR_CONTROL_SERVICE_URL
155+ --repo_env=CC=$(which clang)
156+ --define=using_clang=true
157+ )
158+ echo "BAZEL_FLAGS=${BAZEL_FLAGS[@] }" >> "${GITHUB_ENV}"
157159
158160 - name : EnzymeJaX build
159161 working-directory : ${{ env.BASE_DIR }}/'Enzyme-JaX'
0 commit comments