34 lines
1.1 KiB
Bash
34 lines
1.1 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
|
||
|
|
set -euo pipefail
|
||
|
|
ci_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
|
|
|
||
|
|
source "$ci_dir/pyenv_helper.sh"
|
||
|
|
|
||
|
|
# Parse common arguments
|
||
|
|
source "$ci_dir/util/python/common_arg_parser.sh"
|
||
|
|
parse_python_args "$@"
|
||
|
|
# Pin cuda-toolkit to the container's CTK minor and set cuda_version /
|
||
|
|
# cuda_major_version (-ctk-mode latest opts out). See pyenv_helper.sh.
|
||
|
|
pin_cuda_toolkit "${ctk_mode}"
|
||
|
|
|
||
|
|
# Setup Python environment
|
||
|
|
setup_python_env "${py_version}"
|
||
|
|
|
||
|
|
# Fetch or build the cuda_cccl wheel:
|
||
|
|
if [[ -n "${GITHUB_ACTIONS:-}" ]]; then
|
||
|
|
wheel_artifact_name=$("$ci_dir/util/workflow/get_wheel_artifact_name.sh")
|
||
|
|
"$ci_dir/util/artifacts/download.sh" "${wheel_artifact_name}" /home/coder/cccl/
|
||
|
|
else
|
||
|
|
"$ci_dir/build_cuda_cccl_python.sh" -py-version "${py_version}"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Install cuda_cccl
|
||
|
|
CUDA_CCCL_WHEEL_PATH="$(ls /home/coder/cccl/wheelhouse/cuda_cccl-*.whl)"
|
||
|
|
ctk_flavor="$(ctk_extra_flavor "${ctk_mode}")"
|
||
|
|
python -m pip install "${CUDA_CCCL_WHEEL_PATH}[test-${ctk_flavor}${cuda_major_version}]"
|
||
|
|
|
||
|
|
# Run tests for core package
|
||
|
|
cd "/home/coder/cccl/python/cuda_cccl/tests/"
|
||
|
|
python -m pytest -n auto -v headers/
|