From 9aaa21f80e0bf2f29709e2c7f9ea247c1dd89080 Mon Sep 17 00:00:00 2001
From: brandon-b-miller <brmiller@nvidia.com>
Date: Wed, 9 Oct 2024 08:23:41 -0700
Subject: [PATCH] ship makefile, find and build tests

---
 ci/build_tests.sh | 14 ++++++++++++++
 ci/test_conda.sh  |  1 +
 ci/test_wheel.sh  |  1 +
 pyproject.toml    |  2 +-
 4 files changed, 17 insertions(+), 1 deletion(-)
 create mode 100755 ci/build_tests.sh

diff --git a/ci/build_tests.sh b/ci/build_tests.sh
new file mode 100755
index 0000000..8bf6fe3
--- /dev/null
+++ b/ci/build_tests.sh
@@ -0,0 +1,14 @@
+#!/bin/bash
+# Copyright (c) 2024, NVIDIA CORPORATION
+
+PY_SCRIPT="
+import numba_cuda
+root = numba_cuda.__file__.rstrip('__init__.py')
+test_dir = root + numba/cuda/tests/test_binary_generation/
+print(test_dir)
+"
+
+TEST_DIR=$(python -c "$PY_SCRIPT")
+pushd $TEST_DIR
+make
+popd
diff --git a/ci/test_conda.sh b/ci/test_conda.sh
index 4002e67..200e91f 100755
--- a/ci/test_conda.sh
+++ b/ci/test_conda.sh
@@ -58,6 +58,7 @@ if [ "$USE_PYNVJITLINK" == true ]; then
     set +u
     conda install -c rapidsai pynvjitlink
     set -u
+    sh build_tests.sh
 fi
 
 rapids-logger "Run Tests"
diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
index b6ee2d1..a69b801 100755
--- a/ci/test_wheel.sh
+++ b/ci/test_wheel.sh
@@ -19,6 +19,7 @@ python -m pip install \
 if [ "$USE_PYNVJITLINK" == true ]; then
     rapids-logger "Install pynvjitlink"
     python -m pip install pynvjitlink-cu12
+    sh build_tests.sh
 fi
 
 rapids-logger "Install wheel"
diff --git a/pyproject.toml b/pyproject.toml
index 4f2291b..e7aefdb 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -36,4 +36,4 @@ license-files = ["LICENSE"]
 include = ["numba_cuda*"]
 
 [tool.setuptools.package-data]
-"*" = ["*.cu", "*.h", "*.hpp", "*.ptx", "VERSION"]
+"*" = ["*.cu", "*.h", "*.hpp", "*.ptx", "VERSION", "Makefile"]