@@ -62,23 +62,23 @@ jobs:
62
62
id : cache-llvm-source
63
63
uses : actions/cache@v4
64
64
with :
65
- path : mlir/llvm-project
65
+ path : ${{ github.workspace }}/ mlir/llvm-project
66
66
key : llvm-${{ needs.constants.outputs.llvm_version }}-default-source
67
67
enableCrossOsArchive : True
68
68
69
69
- name : Cache MHLO Source
70
70
id : cache-mhlo-source
71
71
uses : actions/cache@v4
72
72
with :
73
- path : mlir/mlir-hlo
73
+ path : ${{ github.workspace }}/ mlir/mlir-hlo
74
74
key : mhlo-${{ needs.constants.outputs.mhlo_version }}-default-source
75
75
enableCrossOsArchive : True
76
76
77
77
- name : Cache Enzyme Source
78
78
id : cache-enzyme-source
79
79
uses : actions/cache@v4
80
80
with :
81
- path : mlir/Enzyme
81
+ path : ${{ github.workspace }}/ mlir/Enzyme
82
82
key : enzyme-${{ needs.constants.outputs.enzyme_version }}-default-source
83
83
enableCrossOsArchive : True
84
84
@@ -88,60 +88,48 @@ jobs:
88
88
with :
89
89
repository : llvm/llvm-project
90
90
ref : ${{ needs.constants.outputs.llvm_version }}
91
- path : mlir/llvm-project
91
+ path : ${{ github.workspace }}/ mlir/llvm-project
92
92
93
93
- name : Clone MHLO Submodule
94
94
if : steps.cache-mhlo-source.outputs.cache-hit != 'true'
95
95
uses : actions/checkout@v4
96
96
with :
97
97
repository : tensorflow/mlir-hlo
98
98
ref : ${{ needs.constants.outputs.mhlo_version }}
99
- path : mlir/mlir-hlo
99
+ path : ${{ github.workspace }}/ mlir/mlir-hlo
100
100
101
101
- name : Clone Enzyme Submodule
102
102
if : steps.cache-enzyme-source.outputs.cache-hit != 'true'
103
103
uses : actions/checkout@v4
104
104
with :
105
105
repository : EnzymeAD/Enzyme
106
106
ref : ${{ needs.constants.outputs.enzyme_version }}
107
- path : mlir/Enzyme
107
+ path : ${{ github.workspace }}/ mlir/Enzyme
108
108
109
109
# Cache external project builds
110
110
- name : Restore LLVM Build
111
111
id : cache-llvm-build
112
112
uses : actions/cache/restore@v4
113
113
with :
114
- path : llvm-build
114
+ path : ${{ github.workspace }}/ llvm-build
115
115
key : ${{ matrix.container_name }}-llvm-${{ needs.constants.outputs.llvm_version }}-wheel-build
116
116
117
117
- name : Restore MHLO Build
118
118
id : cache-mhlo-build
119
119
uses : actions/cache/restore@v4
120
120
with :
121
- path : mhlo-build
121
+ path : ${{ github.workspace }}/ mhlo-build
122
122
key : ${{ matrix.container_name }}-mhlo-${{ needs.constants.outputs.mhlo_version }}-wheel-build
123
123
lookup-only : True
124
124
125
125
- name : Restore Enzyme Build
126
126
id : cache-enzyme-build
127
127
uses : actions/cache/restore@v4
128
128
with :
129
- path : enzyme-build
129
+ path : ${{ github.workspace }}/ enzyme-build
130
130
key : ${{ matrix.container_name }}-enzyme-${{ needs.constants.outputs.llvm_version }}-${{ needs.constants.outputs.enzyme_version }}-wheel-build
131
131
lookup-only : True
132
132
133
- - name : Build LLD
134
- if : steps.cache-llvm-build.outputs.cache-hit != 'true'
135
- run : |
136
- set -x
137
- # With GCC 13, LLVM fails some tests, then we use GCC 12 instead (Copied from below)
138
- export GCC_VERSION=12
139
- docker run --rm --platform linux/aarch64 \
140
- -v /var/run/docker.sock:/var/run/docker.sock \
141
- -v `pwd`:/catalyst \
142
- -i ${{ matrix.container_img }} \
143
- bash /catalyst/.github/workflows/scripts/linux_arm64/rh8/build_lld.sh $GCC_VERSION ${{ matrix.python_version.major_minor }} ${{ matrix.python_version.patch }} ${{ matrix.python_version.package }}
144
-
145
133
- name : Build LLVM / MLIR
146
134
if : steps.cache-llvm-build.outputs.cache-hit != 'true'
147
135
run : |
@@ -159,7 +147,7 @@ jobs:
159
147
if : steps.cache-llvm-build.outputs.cache-hit != 'true'
160
148
uses : actions/cache/save@v4
161
149
with :
162
- path : llvm-build
150
+ path : ${{ github.workspace }}/ llvm-build
163
151
key : ${{ matrix.container_name }}-llvm-${{ needs.constants.outputs.llvm_version }}-wheel-build
164
152
165
153
- name : Build MHLO Dialect
@@ -178,7 +166,7 @@ jobs:
178
166
if : steps.cache-mhlo-build.outputs.cache-hit != 'true'
179
167
uses : actions/cache/save@v4
180
168
with :
181
- path : mhlo-build
169
+ path : ${{ github.workspace }}/ mhlo-build
182
170
key : ${{ matrix.container_name }}-mhlo-${{ needs.constants.outputs.mhlo_version }}-wheel-build
183
171
184
172
- name : Build Enzyme
@@ -197,7 +185,7 @@ jobs:
197
185
if : steps.cache-enzyme-build.outputs.cache-hit != 'true'
198
186
uses : actions/cache/save@v4
199
187
with :
200
- path : enzyme-build
188
+ path : ${{ github.workspace }}/ enzyme-build
201
189
key : ${{ matrix.container_name }}-enzyme-${{ needs.constants.outputs.llvm_version }}-${{ needs.constants.outputs.enzyme_version }}-wheel-build
202
190
203
191
catalyst-linux-wheels-arm64 :
@@ -230,7 +218,7 @@ jobs:
230
218
id : cache-llvm-source
231
219
uses : actions/cache/restore@v4
232
220
with :
233
- path : mlir/llvm-project
221
+ path : ${{ github.workspace }}/ mlir/llvm-project
234
222
key : llvm-${{ needs.constants.outputs.llvm_version }}-default-source
235
223
enableCrossOsArchive : True
236
224
fail-on-cache-miss : True
@@ -239,15 +227,15 @@ jobs:
239
227
id : cache-llvm-build
240
228
uses : actions/cache/restore@v4
241
229
with :
242
- path : llvm-build
230
+ path : ${{ github.workspace }}/ llvm-build
243
231
key : ${{ matrix.container_name }}-llvm-${{ needs.constants.outputs.llvm_version }}-wheel-build
244
232
fail-on-cache-miss : True
245
233
246
234
- name : Get Cached MHLO Source
247
235
id : cache-mhlo-source
248
236
uses : actions/cache/restore@v4
249
237
with :
250
- path : mlir/mlir-hlo
238
+ path : ${{ github.workspace }}/ mlir/mlir-hlo
251
239
key : mhlo-${{ needs.constants.outputs.mhlo_version }}-default-source
252
240
enableCrossOsArchive : True
253
241
fail-on-cache-miss : True
@@ -256,15 +244,15 @@ jobs:
256
244
id : cache-mhlo-build
257
245
uses : actions/cache/restore@v4
258
246
with :
259
- path : mhlo-build
247
+ path : ${{ github.workspace }}/ mhlo-build
260
248
key : ${{ matrix.container_name }}-mhlo-${{ needs.constants.outputs.mhlo_version }}-wheel-build
261
249
fail-on-cache-miss : True
262
250
263
251
- name : Get Cached Enzyme Source
264
252
id : cache-enzyme-source
265
253
uses : actions/cache/restore@v4
266
254
with :
267
- path : mlir/Enzyme
255
+ path : ${{ github.workspace }}/ mlir/Enzyme
268
256
key : enzyme-${{ needs.constants.outputs.enzyme_version }}-default-source
269
257
enableCrossOsArchive : True
270
258
fail-on-cache-miss : True
@@ -273,7 +261,7 @@ jobs:
273
261
id : cache-enzyme-build
274
262
uses : actions/cache/restore@v4
275
263
with :
276
- path : enzyme-build
264
+ path : ${{ github.workspace }}/ enzyme-build
277
265
key : ${{ matrix.container_name }}-enzyme-${{ needs.constants.outputs.llvm_version }}-${{ needs.constants.outputs.enzyme_version }}-wheel-build
278
266
fail-on-cache-miss : True
279
267
@@ -292,7 +280,7 @@ jobs:
292
280
uses : actions/upload-artifact@v4
293
281
with :
294
282
name : catalyst-linux_arm64-wheel-py-${{ matrix.python_version.major_minor}}.zip
295
- path : wheel/
283
+ path : ${{ github.workspace }}/ wheel/
296
284
retention-days : 14
297
285
298
286
test-wheels :
@@ -326,16 +314,7 @@ jobs:
326
314
uses : actions/download-artifact@v4
327
315
with :
328
316
name : catalyst-linux_arm64-wheel-py-${{ matrix.python_version.major_minor }}.zip
329
- path : dist
330
-
331
- # Needed for accessing llvm-symbolizer
332
- - name : Get Cached LLVM Build
333
- id : cache-llvm-build
334
- uses : actions/cache@v4
335
- with :
336
- path : llvm-build
337
- key : ${{ matrix.container_name }}-llvm-${{ needs.constants.outputs.llvm_version }}-wheel-build
338
- fail-on-cache-miss : True
317
+ path : ${{ github.workspace }}/dist
339
318
340
319
- name : Run Python Pytest Tests
341
320
run : |
0 commit comments