From f5968e189debad1e223c80ac65632ce13c0088a9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gr=C3=A9goire=20KUBLER?= <gregoire.kubler@proton.me>
Date: Fri, 6 Sep 2024 15:02:44 +0200
Subject: [PATCH] fix : impossible conversion from float ** to void **
---
src/operator/AddImpl.cpp | 2 +-
src/operator/SubImpl.cpp | 4 ++--
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/operator/AddImpl.cpp b/src/operator/AddImpl.cpp
index c6f8a3e..de7ea92 100644
--- a/src/operator/AddImpl.cpp
+++ b/src/operator/AddImpl.cpp
@@ -195,7 +195,7 @@ void Aidge::AddImpl_cuda::backward_(const Tensor& outputGrad, const std::vector<
tensorDesc,
&workspaceSize));
- float *d_workspace;
+ void *d_workspace;
CHECK_CUDA_STATUS(cudaMalloc(&d_workspace, workspaceSize));
CHECK_CUDNN_STATUS(cudnnReduceTensor(CudaContext::cudnnHandle(),
diff --git a/src/operator/SubImpl.cpp b/src/operator/SubImpl.cpp
index 3fcf6e7..a04a1c3 100644
--- a/src/operator/SubImpl.cpp
+++ b/src/operator/SubImpl.cpp
@@ -201,7 +201,7 @@ void Aidge::SubImpl_cuda::backward_(const Tensor& outputGrad, const std::vector<
tensorDesc,
&workspaceSize));
- float *d_workspace;
+ void *d_workspace;
CHECK_CUDA_STATUS(cudaMalloc(&d_workspace, workspaceSize));
CHECK_CUDNN_STATUS(cudnnReduceTensor(CudaContext::cudnnHandle(),
@@ -220,4 +220,4 @@ void Aidge::SubImpl_cuda::backward_(const Tensor& outputGrad, const std::vector<
CHECK_CUDNN_STATUS(cudnnDestroyTensorDescriptor(tensorDesc));
}
}
-}
\ No newline at end of file
+}
--
GitLab