diff --git a/cpp/neuralnet/cudabackend.cpp b/cpp/neuralnet/cudabackend.cpp index b91ded728..6657f20b6 100644 --- a/cpp/neuralnet/cudabackend.cpp +++ b/cpp/neuralnet/cudabackend.cpp @@ -2237,7 +2237,7 @@ struct Buffers { CUDA_ERR("Buffers",cudaMalloc(reinterpret_cast(&inputGlobalBufFloat), inputGlobalBufBytesFloat)); CUDA_ERR("Buffers",cudaMalloc(&inputGlobalBuf, inputGlobalBufBytes)); if(m.numInputMetaChannels > 0) { - CUDA_ERR("Buffers",cudaMalloc(&inputMetaBufFloat, inputMetaBufBytesFloat)); + CUDA_ERR("Buffers",cudaMalloc(reinterpret_cast(&inputMetaBufFloat), inputMetaBufBytesFloat)); CUDA_ERR("Buffers",cudaMalloc(&inputMetaBuf, inputMetaBufBytes)); } else {