Fix cudaMalloc conversion from T** to void**

CEA-LIST · May 21, 2024 · 037fa27 · 037fa27
1 parent 2e553a3
commit 037fa27
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/include/containers/CudaTensor.hpp b/include/containers/CudaTensor.hpp
@@ -593,7 +593,7 @@ T* N2D2::CudaDeviceTensor<T>::getDevicePtr(int dev) const
 
         if (mDataDevice[dev] == NULL) {
             // Lazy memory allocation
-            CHECK_CUDA_STATUS(cudaMalloc(&(mDataDevice[dev]),
+            CHECK_CUDA_STATUS(cudaMalloc(reinterpret_cast<void**>(&(mDataDevice[dev])),
                                          mCudaBaseTensor.size() * sizeof(T)));
         }
 
@@ -782,7 +782,7 @@ void N2D2::CudaDeviceTensor<T>::aggregate(int srcDev, int dstDev) const
     if (mForeignDataDevice[dstDev] == NULL) {
 		// Lazy allocation
 		CHECK_CUDA_STATUS(cudaMalloc(
-		&mForeignDataDevice[dstDev], mCudaBaseTensor.size() * sizeof(T)));
+		reinterpret_cast<void**>(&mForeignDataDevice[dstDev]), mCudaBaseTensor.size() * sizeof(T)));
     }
 
     CHECK_CUDA_STATUS(cudaMemcpyPeer(