Skip to content

Commit 8ba2967

Browse files
johncalespjohncalesp
authored
changed env variable name for build process (#48)
Co-authored-by: John Calderon <john.calderon@centml.ai>
1 parent 58f243a commit 8ba2967

File tree

2 files changed

+24
-24
lines changed

2 files changed

+24
-24
lines changed

‎.github/workflows/whl-build-all.yaml‎

Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -48,28 +48,28 @@ jobs:
4848
run: |
4949
./build_scripts/build_wheel.sh python3.8
5050
env:
51-
CUDA_TAG: cu113
51+
VERSION_CUDA_TAG: cu113
5252
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99
5353

5454
- name: Build Python3.9 wheel
5555
run: |
5656
./build_scripts/build_wheel.sh python3.9
5757
env:
58-
CUDA_TAG: cu113
58+
VERSION_CUDA_TAG: cu113
5959
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99
6060

6161
- name: Build Python3.10 wheel
6262
run: |
6363
./build_scripts/build_wheel.sh python3.10
6464
env:
65-
CUDA_TAG: cu113
65+
VERSION_CUDA_TAG: cu113
6666
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99
6767

6868
- name: Build Python3.11 wheel
6969
run: |
7070
./build_scripts/build_wheel.sh python3.11
7171
env:
72-
CUDA_TAG: cu113
72+
VERSION_CUDA_TAG: cu113
7373
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99
7474

7575
- name: Upload wheels to S3
@@ -111,28 +111,28 @@ jobs:
111111
run: |
112112
./build_scripts/build_wheel.sh python3.8
113113
env:
114-
CUDA_TAG: cu116
114+
VERSION_CUDA_TAG: cu116
115115
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99
116116

117117
- name: Build Python3.9 wheel
118118
run: |
119119
./build_scripts/build_wheel.sh python3.9
120120
env:
121-
CUDA_TAG: cu116
121+
VERSION_CUDA_TAG: cu116
122122
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99
123123

124124
- name: Build Python3.10 wheel
125125
run: |
126126
./build_scripts/build_wheel.sh python3.10
127127
env:
128-
CUDA_TAG: cu116
128+
VERSION_CUDA_TAG: cu116
129129
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99
130130

131131
- name: Build Python3.11 wheel
132132
run: |
133133
./build_scripts/build_wheel.sh python3.11
134134
env:
135-
CUDA_TAG: cu116
135+
VERSION_CUDA_TAG: cu116
136136
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99
137137

138138
- name: Upload wheels to S3
@@ -174,28 +174,28 @@ jobs:
174174
run: |
175175
./build_scripts/build_wheel.sh python3.8
176176
env:
177-
CUDA_TAG: cu117
177+
VERSION_CUDA_TAG: cu117
178178
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99
179179

180180
- name: Build Python3.9 wheel
181181
run: |
182182
./build_scripts/build_wheel.sh python3.9
183183
env:
184-
CUDA_TAG: cu117
184+
VERSION_CUDA_TAG: cu117
185185
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99
186186

187187
- name: Build Python3.10 wheel
188188
run: |
189189
./build_scripts/build_wheel.sh python3.10
190190
env:
191-
CUDA_TAG: cu117
191+
VERSION_CUDA_TAG: cu117
192192
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99
193193

194194
- name: Build Python3.11 wheel
195195
run: |
196196
./build_scripts/build_wheel.sh python3.11
197197
env:
198-
CUDA_TAG: cu117
198+
VERSION_CUDA_TAG: cu117
199199
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.7.101,nvidia-cuda-runtime-cu11==11.7.99
200200

201201
- name: Upload wheels to S3
@@ -237,28 +237,28 @@ jobs:
237237
run: |
238238
./build_scripts/build_wheel.sh python3.8
239239
env:
240-
CUDA_TAG: cu118
240+
VERSION_CUDA_TAG: cu118
241241
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.8.87,nvidia-cuda-runtime-cu11==11.8.89
242242

243243
- name: Build Python3.9 wheel
244244
run: |
245245
./build_scripts/build_wheel.sh python3.9
246246
env:
247-
CUDA_TAG: cu118
247+
VERSION_CUDA_TAG: cu118
248248
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.8.87,nvidia-cuda-runtime-cu11==11.8.89
249249

250250
- name: Build Python3.10 wheel
251251
run: |
252252
./build_scripts/build_wheel.sh python3.10
253253
env:
254-
CUDA_TAG: cu118
254+
VERSION_CUDA_TAG: cu118
255255
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.8.87,nvidia-cuda-runtime-cu11==11.8.89
256256

257257
- name: Build Python3.11 wheel
258258
run: |
259259
./build_scripts/build_wheel.sh python3.11
260260
env:
261-
CUDA_TAG: cu118
261+
VERSION_CUDA_TAG: cu118
262262
EXTRA_REQUIRES: nvidia-cuda-cupti-cu11==11.8.87,nvidia-cuda-runtime-cu11==11.8.89
263263

264264
- name: Upload wheels to S3
@@ -300,25 +300,25 @@ jobs:
300300
run: |
301301
./build_scripts/build_wheel.sh python3.8
302302
env:
303-
CUDA_TAG: cu121
303+
VERSION_CUDA_TAG: cu121
304304

305305
- name: Build Python3.9 wheel
306306
run: |
307307
./build_scripts/build_wheel.sh python3.9
308308
env:
309-
CUDA_TAG: cu121
309+
VERSION_CUDA_TAG: cu121
310310

311311
- name: Build Python3.10 wheel
312312
run: |
313313
./build_scripts/build_wheel.sh python3.10
314314
env:
315-
CUDA_TAG: cu121
315+
VERSION_CUDA_TAG: cu121
316316

317317
- name: Build Python3.11 wheel
318318
run: |
319319
./build_scripts/build_wheel.sh python3.11
320320
env:
321-
CUDA_TAG: cu121
321+
VERSION_CUDA_TAG: cu121
322322

323323
- name: Upload wheels to S3
324324
run: |

‎analyzer/setup.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
We define two additional environment arguments during build to include dependencies for
1818
different versions of CUDA we're targeting.
1919
20-
1. CUDA_TAG : This follows the "version number". For example, if version="1.0" and CUDA_TAG =cu121, then the
20+
1. VERSION_CUDA_TAG : This follows the "version number". For example, if version="1.0" and VERSION_CUDA_TAG =cu121, then the
2121
version we pass into setuptools.setup would be "1.0+cu121"
2222
2323
2. EXTRA_REQUIRES: Depends on the CUDA version, we need additional pip libraries to provide CUPTI (among other things).
@@ -44,8 +44,8 @@
4444
"incremental"
4545
]
4646

47-
CUDA_TAG = os.getenv("CUDA_TAG", default="")
48-
if CUDA_TAG : CUDA_TAG = "+" + CUDA_TAG
47+
VERSION_CUDA_TAG = os.getenv("VERSION_CUDA_TAG", default="")
48+
if VERSION_CUDA_TAG : VERSION_CUDA_TAG = "+" + VERSION_CUDA_TAG
4949
EXTRA_REQUIRES = os.getenv("EXTRA_REQUIRES", default="nvidia-cuda-cupti-cu12,nvidia-cuda-runtime-cu12").split(",")
5050

5151
PACKAGE_DATA = {
@@ -149,7 +149,7 @@ def find_meta(meta):
149149
if __name__ == "__main__":
150150
setup(
151151
name=NAME,
152-
version=VERSION + CUDA_TAG ,
152+
version=VERSION + VERSION_CUDA_TAG ,
153153
description=find_meta("description"),
154154
license=find_meta("license"),
155155
author=find_meta("author"),

0 commit comments

Comments
 (0)