diff --git a/src/operator/AddImpl.cpp b/src/operator/AddImpl.cpp
index c6f8a3eb00ddd52d5b255bba2ebcceb53a9e1a97..de7ea925554906ea5fe1e5dcba268b17a06a47bd 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 3fcf6e7808221bbcaf1fa7a88328e8326be7203c..a04a1c3018b0c9ba455d21ba563253eb3e004e10 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
+}