From 201aee96d9f327b7ec8b965ba89970831f092a20 Mon Sep 17 00:00:00 2001
From: Alexandria Barghi <abarghi@nvidia.com>
Date: Fri, 11 Oct 2024 15:55:37 -0700
Subject: [PATCH] install pylibwholegraph

---
 ci/test_wheel_pylibwholegraph.sh | 38 ++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 ci/test_wheel_pylibwholegraph.sh

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 \
+  .