update

Signed-off-by: minmingzhu <[email protected]>
minmingzhu · Jan 9, 2024 · 1ea9cd6 · 1ea9cd6
1 parent b832f20
commit 1ea9cd6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mllib-dal/src/main/native/PCAImpl.cpp b/mllib-dal/src/main/native/PCAImpl.cpp
@@ -204,7 +204,7 @@ static void doPCAOneAPICompute(
 
     const auto cov_desc =
         covariance_gpu::descriptor<GpuAlgorithmFPType>{}.set_result_options(
-            covariance_gpu::result_options::cor_matrix);
+            covariance_gpu::result_options::cov_matrix);
     t1 = std::chrono::high_resolution_clock::now();
     const auto result = preview::compute(comm, cov_desc, htable);
     t2 = std::chrono::high_resolution_clock::now();
@@ -222,7 +222,7 @@ static void doPCAOneAPICompute(
         const auto pca_desc = descriptor_t().set_deterministic(true);
 
         const auto result_train =
-            train(queue, pca_desc, result.get_cor_matrix());
+            preview::train(comm, pca_desc, result.get_cov_matrix());
         t2 = std::chrono::high_resolution_clock::now();
         duration = (float)std::chrono::duration_cast<std::chrono::milliseconds>(
                        t2 - t1)