diff --git a/ci/test_wheel_pylibwholegraph.sh b/ci/test_wheel_pylibwholegraph.sh new file mode 100644 index 0000000..bc44ac9 --- /dev/null +++ b/ci/test_wheel_pylibwholegraph.sh @@ -0,0 +1,38 @@ +#!/bin/bash +# Copyright (c) 2023-2024, NVIDIA CORPORATION. + +set -e # abort the script on error +set -o pipefail # piped commands propagate their error +set -E # ERR traps are inherited by subcommands + +mkdir -p ./dist +RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})" +RAPIDS_PY_WHEEL_NAME="pylibwholegraph_${RAPIDS_PY_CUDA_SUFFIX}" rapids-download-wheels-from-s3 ./local-deps + +PKG_CUDA_VER="$(echo ${CUDA_VERSION} | cut -d '.' -f1,2 | tr -d '.')" +PKG_CUDA_VER_MAJOR=${PKG_CUDA_VER:0:2} +if [[ "${PKG_CUDA_VER_MAJOR}" == "12" ]]; then + INDEX_URL="https://download.pytorch.org/whl/cu121" +else + INDEX_URL="https://download.pytorch.org/whl/cu${PKG_CUDA_VER}" +fi +RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${PWD}/test-results"} +RAPIDS_COVERAGE_DIR=${RAPIDS_COVERAGE_DIR:-"${PWD}/coverage-results"} +mkdir -p "${RAPIDS_TESTS_DIR}" "${RAPIDS_COVERAGE_DIR}" + +# echo to expand wildcard before adding `[extra]` requires for pip +rapids-logger "Installing Packages" +rapids-retry python -m pip install \ + --extra-index-url ${INDEX_URL} \ + $(echo ./dist/pylibwholegraph*.whl) \ + pytest \ + pytest-forked \ + numpy + +rapids-logger "pytest pylibwholegraph" +cd python/pylibwholegraph/pylibwholegraph/tests +python -m pytest \ + --cache-clear \ + --forked \ + --import-mode=append \ + .