From 9d4c014f608e344f3cf7b891c80675235758f736 Mon Sep 17 00:00:00 2001
From: Rutwik Choughule <rutwik.choughule@gmail.com>
Date: Tue, 25 Jan 2022 17:22:31 +0530
Subject: [PATCH] update sample cudaNvSci

---
 Samples/4_CUDA_Libraries/cudaNvSci/cudaNvSci.cpp   | 9 +++++++++
 Samples/4_CUDA_Libraries/cudaNvSci/imageKernels.cu | 2 +-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/Samples/4_CUDA_Libraries/cudaNvSci/cudaNvSci.cpp b/Samples/4_CUDA_Libraries/cudaNvSci/cudaNvSci.cpp
index 77607dce..988df3d4 100644
--- a/Samples/4_CUDA_Libraries/cudaNvSci/cudaNvSci.cpp
+++ b/Samples/4_CUDA_Libraries/cudaNvSci/cudaNvSci.cpp
@@ -113,7 +113,12 @@ class cudaNvSciSignal {
         "%d.%d\n\n",
         m_cudaDeviceId, _ConvertSMVer2ArchName(major, minor), major, minor);
 
+#ifdef cuDeviceGetUuid_v2
+    CUresult res = cuDeviceGetUuid_v2(&m_devUUID, m_cudaDeviceId);
+#else
     CUresult res = cuDeviceGetUuid(&m_devUUID, m_cudaDeviceId);
+#endif
+
     if (res != CUDA_SUCCESS) {
       fprintf(stderr, "Driver API error = %04d \n", res);
       exit(EXIT_FAILURE);
@@ -387,7 +392,11 @@ class cudaNvSciWait {
     checkCudaErrors(cudaSetDevice(m_cudaDeviceId));
     checkCudaErrors(
         cudaStreamCreateWithFlags(&streamToRun, cudaStreamNonBlocking));
+#ifdef cuDeviceGetUuid_v2
+    CUresult res = cuDeviceGetUuid_v2(&m_devUUID, m_cudaDeviceId);
+#else
     CUresult res = cuDeviceGetUuid(&m_devUUID, m_cudaDeviceId);
+#endif
     if (res != CUDA_SUCCESS) {
       fprintf(stderr, "Driver API error = %04d \n", res);
       exit(EXIT_FAILURE);
diff --git a/Samples/4_CUDA_Libraries/cudaNvSci/imageKernels.cu b/Samples/4_CUDA_Libraries/cudaNvSci/imageKernels.cu
index 528fc328..11075eb1 100644
--- a/Samples/4_CUDA_Libraries/cudaNvSci/imageKernels.cu
+++ b/Samples/4_CUDA_Libraries/cudaNvSci/imageKernels.cu
@@ -93,7 +93,7 @@ void launchGrayScaleKernel(unsigned int *d_rgbaImage,
       d_rgbaImage, imageWidth, imageHeight);
 
   unsigned int *outputData;
-  checkCudaErrors(cudaMallocHost(&outputData, sizeof(unsigned int) * imageWidth * imageHeight));
+  checkCudaErrors(cudaMallocHost((void**)&outputData, sizeof(unsigned int) * imageWidth * imageHeight));
   checkCudaErrors(cudaMemcpyAsync(
       outputData, d_rgbaImage, sizeof(unsigned int) * imageWidth * imageHeight,
       cudaMemcpyDeviceToHost, stream));