blob: a942b206f3ed3b74dc5ac44317b08e23186bb796 (
plain)
1
2
3
4
5
6
7
8
9
10
11
|
--- cpp/neuralnet/cudabackend.cpp
+++ cpp/neuralnet/cudabackend.cpp
@@ -2237,7 +2237,7 @@
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 {
|