diff --git a/cpp/include/raft/linalg/detail/gemm.hpp b/cpp/include/raft/linalg/detail/gemm.hpp index 29308304d8..50a8be6018 100644 --- a/cpp/include/raft/linalg/detail/gemm.hpp +++ b/cpp/include/raft/linalg/detail/gemm.hpp @@ -61,7 +61,7 @@ void gemm(const raft::handle_t& handle, const math_t* B, const int ldb, const math_t* beta, - const math_t* C, + math_t* C, const int ldc, cudaStream_t stream) { diff --git a/cpp/include/raft/linalg/gemm.cuh b/cpp/include/raft/linalg/gemm.cuh index 9670834ff0..16a5bc48ea 100644 --- a/cpp/include/raft/linalg/gemm.cuh +++ b/cpp/include/raft/linalg/gemm.cuh @@ -58,7 +58,7 @@ void gemm(const raft::handle_t& handle, const math_t* B, const int ldb, const math_t* beta, - const math_t* C, + math_t* C, const int ldc, cudaStream_t stream) { diff --git a/cpp/include/raft/linalg/gemm.hpp b/cpp/include/raft/linalg/gemm.hpp index 56621e4f8b..37c6b2d552 100644 --- a/cpp/include/raft/linalg/gemm.hpp +++ b/cpp/include/raft/linalg/gemm.hpp @@ -63,7 +63,7 @@ void gemm(const raft::handle_t& handle, const math_t* B, const int ldb, const math_t* beta, - const math_t* C, + math_t* C, const int ldc, cudaStream_t stream) { @@ -181,4 +181,4 @@ void gemm(const raft::handle_t& handle, } // end namespace linalg } // end namespace raft -#endif \ No newline at end of file +#endif