mirror of
https://github.com/NVIDIA/nvbench.git
synced 2026-04-20 14:58:54 +00:00
54 lines
2.0 KiB
Bash
Executable File
54 lines
2.0 KiB
Bash
Executable File
#!/bin/bash
|
|
set -euo pipefail
|
|
|
|
# Target script for `docker run` command in test_pynvbench.sh
|
|
# The /workspace pathnames are hard-wired here.
|
|
|
|
# Install GCC 13 toolset (needed for builds that might happen during testing)
|
|
/workspace/ci/util/retry.sh 5 30 dnf -y install gcc-toolset-13-gcc gcc-toolset-13-gcc-c++
|
|
echo -e "#!/bin/bash\nsource /opt/rh/gcc-toolset-13/enable" >/etc/profile.d/enable_devtools.sh
|
|
source /etc/profile.d/enable_devtools.sh
|
|
|
|
# Install NVIDIA Management Library (required by pynvbench at runtime)
|
|
# This is a driver library that we exclude from the wheel but need for testing
|
|
echo "Installing libnvidia-ml.so.1..."
|
|
/workspace/ci/util/retry.sh 5 30 dnf -y install nvidia-driver-libs
|
|
|
|
# Set up Python environment (only if not already available)
|
|
source /workspace/ci/pyenv_helper.sh
|
|
if ! command -v python${py_version} &> /dev/null; then
|
|
setup_python_env "${py_version}"
|
|
fi
|
|
|
|
# Upgrade pip
|
|
python -m pip install --upgrade pip
|
|
|
|
echo "Python version: $(python --version)"
|
|
echo "CUDA version: $(nvcc --version | grep release)"
|
|
|
|
# Wheel should be in /workspace/wheelhouse (downloaded by workflow or built locally)
|
|
WHEELHOUSE_DIR="/workspace/wheelhouse"
|
|
|
|
# Find and install pynvbench wheel
|
|
# Look for .cu${cuda_version} in the version string (e.g., pynvbench-0.0.1.dev1+g123.cu12-...)
|
|
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
|
|
|
|
echo "Installing wheel: $PYNVBENCH_WHEEL_PATH"
|
|
python -m pip install "${PYNVBENCH_WHEEL_PATH}[test]"
|
|
|
|
# Capture the Python path before changing directories
|
|
# This prevents .python-version file from interfering
|
|
PYTHON_EXE="$(which python)"
|
|
echo "Using Python: $PYTHON_EXE"
|
|
|
|
# Run tests
|
|
cd "/workspace/python/test/"
|
|
"$PYTHON_EXE" -m pytest -v test_nvbench.py
|