From 2d66d612937efd5e313112f3d6c011c7e29ab3f0 Mon Sep 17 00:00:00 2001
From: achirkin <achirkin@users.noreply.github.com>
Date: Mon, 27 Nov 2023 09:35:59 +0100
Subject: [PATCH] Check the NVTX headers explicitly to make sure the cpu-only
 build succeeds

---
 cpp/bench/ann/CMakeLists.txt | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/cpp/bench/ann/CMakeLists.txt b/cpp/bench/ann/CMakeLists.txt
index 94da8d1104..90ebe4d3a6 100644
--- a/cpp/bench/ann/CMakeLists.txt
+++ b/cpp/bench/ann/CMakeLists.txt
@@ -123,7 +123,12 @@ endif()
 # benchmark cases and iterations. This is to make limited NVTX available to all algos, not just
 # raft.
 if(TARGET CUDA::nvtx3)
-  set(NVTX3_HEADERS_FOUND ON)
+  set(_CMAKE_REQUIRED_INCLUDES_ORIG ${CMAKE_REQUIRED_INCLUDES})
+  get_target_property(CMAKE_REQUIRED_INCLUDES CUDA::nvtx3 INTERFACE_INCLUDE_DIRECTORIES)
+  unset(NVTX3_HEADERS_FOUND CACHE)
+  # Check the headers explicitly to make sure the cpu-only build succeeds
+  CHECK_INCLUDE_FILE_CXX(nvtx3/nvToolsExt.h NVTX3_HEADERS_FOUND)
+  set(CMAKE_REQUIRED_INCLUDES ${_CMAKE_REQUIRED_INCLUDES_ORIG})
 endif()
 
 # ##################################################################################################