From 9b550767978e5fe50057c6bb7f87e9bc7a973542 Mon Sep 17 00:00:00 2001
From: Jeff Hammond <jehammond@nvidia.com>
Date: Mon, 26 Sep 2022 03:41:45 -0700
Subject: [PATCH] accept NVHPC NVC++ as a CUDA compiler when it is so

Signed-off-by: Jeff Hammond <jehammond@nvidia.com>
---
 src/thrust/ThrustStream.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/thrust/ThrustStream.cu b/src/thrust/ThrustStream.cu
index 3a57ab0..f15a392 100644
--- a/src/thrust/ThrustStream.cu
+++ b/src/thrust/ThrustStream.cu
@@ -145,7 +145,7 @@ T ThrustStream<T>::dot()
 #if THRUST_DEVICE_SYSTEM == THRUST_DEVICE_SYSTEM_CUDA || \
     (defined(THRUST_DEVICE_SYSTEM_HIP) && THRUST_DEVICE_SYSTEM_HIP == THRUST_DEVICE_SYSTEM)
 
-#ifdef __NVCC__
+#if defined(__NVCC__) || defined(__NVCOMPILER_CUDA__)
 #define IMPL_FN__(fn) cuda ## fn
 #define IMPL_TYPE__(tpe) cuda ## tpe
 #elif defined(__HIP_PLATFORM_HCC__)