NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-03-14 20:27:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes to test script

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

13
ci/test_pynvbench.sh
View File

@@ -42,20 +42,15 @@ cuda_major_version=$(nvcc --version | grep release | awk '{print $6}' | tr -d ',
# Setup Python environment
setup_python_env "${py_version}"
# Fetch the pynvbench wheel
if [[ -n "${GITHUB_ACTIONS:-}" ]]; then
# In GitHub Actions, wheel is already downloaded to wheelhouse/ by the workflow
WHEELHOUSE_DIR="/workspace/wheelhouse"
else
# For local testing, build the wheel
"$ci_dir/build_pynvbench_wheel.sh" -py-version "${py_version}" -cuda-version "${cuda_version}"
WHEELHOUSE_DIR="/workspace/wheelhouse"
fi
# Wheel should be in /workspace/wheelhouse (downloaded by workflow or built locally)
WHEELHOUSE_DIR="/workspace/wheelhouse"
# Find and install pynvbench wheel
PYNVBENCH_WHEEL_PATH="$(ls ${WHEELHOUSE_DIR}/pynvbench-*+cu${cuda_version}*.whl 2>/dev/null | head -1)"
if [[ -z "$PYNVBENCH_WHEEL_PATH" ]]; then
echo "Error: No pynvbench wheel found in ${WHEELHOUSE_DIR}"
echo "Looking for: pynvbench-*+cu${cuda_version}*.whl"
echo "Contents of ${WHEELHOUSE_DIR}:"
ls -la ${WHEELHOUSE_DIR}/ || true
exit 1
fi