Skip to content
Snippets Groups Projects
Commit f5968e18 authored by Grégoire Kubler's avatar Grégoire Kubler
Browse files

fix : impossible conversion from float ** to void **

parent fcb939c3
No related branches found
No related tags found
1 merge request!38version 0.3.0
Pipeline #54087 passed
......@@ -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(),
......
......@@ -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
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment