diff --git a/cpp/src/pca/pca.cuh b/cpp/src/pca/pca.cuh index fcc83b22b5..5a59ff3db5 100644 --- a/cpp/src/pca/pca.cuh +++ b/cpp/src/pca/pca.cuh @@ -108,7 +108,7 @@ void pcaFit(const raft::handle_t& handle, auto n_components = prms.n_components; if (n_components > prms.n_cols) n_components = prms.n_cols; - raft::stats::mean(mu, input, prms.n_cols, prms.n_rows, true, false, stream); + raft::stats::mean(mu, input, prms.n_cols, prms.n_rows, false, false, stream); auto len = prms.n_cols * prms.n_cols; rmm::device_uvector cov(len, stream); diff --git a/cpp/src/tsvd/tsvd.cuh b/cpp/src/tsvd/tsvd.cuh index 543af045cb..b032b1a69a 100644 --- a/cpp/src/tsvd/tsvd.cuh +++ b/cpp/src/tsvd/tsvd.cuh @@ -271,21 +271,21 @@ void tsvdFitTransform(const raft::handle_t& handle, rmm::device_uvector mu_trans(prms.n_components, stream); raft::stats::mean( - mu_trans.data(), trans_input, prms.n_components, prms.n_rows, true, false, stream); + mu_trans.data(), trans_input, prms.n_components, prms.n_rows, false, false, stream); raft::stats::vars(explained_var, trans_input, mu_trans.data(), prms.n_components, prms.n_rows, - true, + false, false, stream); rmm::device_uvector mu(prms.n_cols, stream); rmm::device_uvector vars(prms.n_cols, stream); - raft::stats::mean(mu.data(), input, prms.n_cols, prms.n_rows, true, false, stream); - raft::stats::vars(vars.data(), input, mu.data(), prms.n_cols, prms.n_rows, true, false, stream); + raft::stats::mean(mu.data(), input, prms.n_cols, prms.n_rows, false, false, stream); + raft::stats::vars(vars.data(), input, mu.data(), prms.n_cols, prms.n_rows, false, false, stream); rmm::device_scalar total_vars(stream); raft::stats::sum(total_vars.data(), vars.data(), std::size_t(1), prms.n_cols, false, stream);