NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-20 06:48:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix wheel naming

Browse Source
This commit is contained in:
Ashwin Srinath
2025-12-03 11:54:46 -05:00
parent 618001143b
commit d1efef03bc
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
ci/build_pynvbench_wheel_inner.sh
View File

@@ -62,15 +62,22 @@ mkdir -p /workspace/wheelhouse
for wheel in dist_repaired/pynvbench-*.whl; do
if [[ -f "$wheel" ]]; then
base_name=$(basename "$wheel" .whl)
# Insert CUDA version before the platform tag
# e.g., pynvbench-0.1.0-cp312-cp312-manylinux_2_28_x86_64.whl
# becomes pynvbench-0.1.0+cu12-cp312-cp312-manylinux_2_28_x86_64.whl
# Append CUDA version to the local version identifier
# e.g., pynvbench-0.1.0.dev1+gabc123-cp312-cp312-manylinux_2_28_x86_64.whl
# becomes pynvbench-0.1.0.dev1+gabc123.cu12-cp312-cp312-manylinux_2_28_x86_64.whl
if [[ "$base_name" =~ ^(.*)-cp([0-9]+)-cp([0-9]+)-(.*) ]]; then
pkg_version="${BASH_REMATCH[1]}"
py_tag="cp${BASH_REMATCH[2]}"
abi_tag="cp${BASH_REMATCH[3]}"
platform="${BASH_REMATCH[4]}"
new_name="${pkg_version}+cu${cuda_major}-${py_tag}-${abi_tag}-${platform}.whl"
# If version has a local part (contains +), append .cu${cuda_major} to it
# Otherwise add +cu${cuda_major}
if [[ "$pkg_version" =~ \+ ]]; then
new_version="${pkg_version}.cu${cuda_major}"
else
new_version="${pkg_version}+cu${cuda_major}"
fi
new_name="${new_version}-${py_tag}-${abi_tag}-${platform}.whl"
mv "$wheel" "/workspace/wheelhouse/${new_name}"
echo "Renamed wheel to: ${new_name}"
else