From 6a0c33dbdf4d889c96af13e0e763d47f2875b003 Mon Sep 17 00:00:00 2001 From: David Wu Date: Sun, 4 Aug 2024 18:46:58 -0400 Subject: [PATCH] Add cast to fix cuda compile issue --- cpp/neuralnet/cudabackend.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 {