Skip to content

Commit 496886d

Browse files
committed
Secondary fixes
1 parent 56d8a4d commit 496886d

2 files changed

Lines changed: 13 additions & 9 deletions

File tree

.github/workflows/test-wheel-linux.yml

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -247,14 +247,10 @@ jobs:
247247
LOCAL_CTK: ${{ matrix.LOCAL_CTK }}
248248
run: run-tests bindings
249249

250-
251-
252250
- name: Ensure cuda-python installable
253251
run: |
254-
echo Showing wheels locally in $PWD
255-
ls *.whl
256252
if [[ "${{ matrix.LOCAL_CTK }}" == 1 ]]; then
257-
pip install cuda_python*.whl --find-links .
253+
pip install cuda_python*.whl
258254
else
259-
pip install $(ls cuda_python*.whl)[all] --find-links .
255+
pip install $(ls cuda_python*.whl)[all]
260256
fi

cuda_python/setup.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,15 +15,23 @@
1515
)
1616

1717

18-
version = Version(version).base_version
18+
base_version = Version(version).base_version
19+
20+
21+
if base_version == version:
22+
# Tagged release
23+
matcher = "~="
24+
else:
25+
# Pre-release version
26+
matcher = "=="
1927

2028

2129
setup(
2230
version=version,
2331
install_requires=[
24-
f"cuda-bindings~={version}",
32+
f"cuda-bindings{matcher}{version}",
2533
],
2634
extras_require={
27-
"all": [f"cuda-bindings[all]~={version}"],
35+
"all": [f"cuda-bindings[all]{matcher}{version}"],
2836
},
2937
)

0 commit comments

Comments
 (0)