Skip to content

Commit 03955f6

Browse files
committed
Improve formatting
1 parent 8427274 commit 03955f6

File tree

1 file changed

+32
-30
lines changed

1 file changed

+32
-30
lines changed

.github/workflows/enzyme-bazel.yml

Lines changed: 32 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -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'
@@ -94,7 +97,6 @@ jobs:
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

Comments
 (0)