update sample cudaNvSci

NVIDIA · Jan 25, 2022 · 7db525e · 7db525e
1 parent b9063ab
commit 7db525e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git 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
@@ -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));