Add cast to fix cuda compile issue

lightvector · Aug 4, 2024 · 6a0c33d · 6a0c33d
1 parent a043874
commit 6a0c33d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/neuralnet/cudabackend.cpp b/cpp/neuralnet/cudabackend.cpp
@@ -2237,7 +2237,7 @@ struct Buffers {
     CUDA_ERR("Buffers",cudaMalloc(reinterpret_cast<void**>(&inputGlobalBufFloat), inputGlobalBufBytesFloat));
     CUDA_ERR("Buffers",cudaMalloc(&inputGlobalBuf, inputGlobalBufBytes));
     if(m.numInputMetaChannels > 0) {
-      CUDA_ERR("Buffers",cudaMalloc(&inputMetaBufFloat, inputMetaBufBytesFloat));
+      CUDA_ERR("Buffers",cudaMalloc(reinterpret_cast<void**>(&inputMetaBufFloat), inputMetaBufBytesFloat));
       CUDA_ERR("Buffers",cudaMalloc(&inputMetaBuf, inputMetaBufBytes));
     }
     else {