From 84f8e150041fc9ce9f8ad5b3f69e538cb83d426e Mon Sep 17 00:00:00 2001
From: Yunsong Wang <yunsongw@nvidia.com>
Date: Fri, 1 Mar 2024 10:59:20 -0800
Subject: [PATCH] Custom rapids-cmake

---
 rapids_config.cmake | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/rapids_config.cmake b/rapids_config.cmake
index 3a88769f6e7..75ca8191c1b 100644
--- a/rapids_config.cmake
+++ b/rapids_config.cmake
@@ -11,6 +11,10 @@
 # or implied. See the License for the specific language governing permissions and limitations under
 # the License.
 # =============================================================================
+
+set(rapids-cmake-repo PointKernel/rapids-cmake)
+set(rapids-cmake-branch bump-nvbench-version)
+
 file(READ "${CMAKE_CURRENT_LIST_DIR}/VERSION" _rapids_version)
 if(_rapids_version MATCHES [[^([0-9][0-9])\.([0-9][0-9])\.([0-9][0-9])]])
   set(RAPIDS_VERSION_MAJOR "${CMAKE_MATCH_1}")