Revert SPMD K-means instantiation to fix #2959 (#2986)

oneapi-src · Nov 22, 2024 · 1bef482 · 1bef482
1 parent 08fd918
commit 1bef482
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 10 deletions.
diff --git a/cpp/oneapi/dal/algo/kmeans/detail/infer_ops_dpc.cpp b/cpp/oneapi/dal/algo/kmeans/detail/infer_ops_dpc.cpp
@@ -42,13 +42,10 @@ struct infer_ops_dispatcher<Policy, Float, Method, Task> {
     template struct ONEDAL_EXPORT                                         \
         infer_ops_dispatcher<dal::detail::spmd_data_parallel_policy, F, M, T>;
 
-#define INSTANTIATE_NON_DISTRIBUTED(F, M, T) \
-    template struct ONEDAL_EXPORT infer_ops_dispatcher<dal::detail::data_parallel_policy, F, M, T>;
-
 INSTANTIATE(float, method::lloyd_dense, task::clustering)
 INSTANTIATE(double, method::lloyd_dense, task::clustering)
-INSTANTIATE_NON_DISTRIBUTED(float, method::lloyd_csr, task::clustering)
-INSTANTIATE_NON_DISTRIBUTED(double, method::lloyd_csr, task::clustering)
+INSTANTIATE(float, method::lloyd_csr, task::clustering)
+INSTANTIATE(double, method::lloyd_csr, task::clustering)
 
 } // namespace v1
 } // namespace oneapi::dal::kmeans::detail
diff --git a/cpp/oneapi/dal/algo/kmeans/detail/train_ops_dpc.cpp b/cpp/oneapi/dal/algo/kmeans/detail/train_ops_dpc.cpp
@@ -43,13 +43,10 @@ struct train_ops_dispatcher<Policy, Float, Method, Task> {
     template struct ONEDAL_EXPORT                                         \
         train_ops_dispatcher<dal::detail::spmd_data_parallel_policy, F, M, T>;
 
-#define INSTANTIATE_NON_DISTRIBUTED(F, M, T) \
-    template struct ONEDAL_EXPORT train_ops_dispatcher<dal::detail::data_parallel_policy, F, M, T>;
-
 INSTANTIATE(float, method::lloyd_dense, task::clustering)
 INSTANTIATE(double, method::lloyd_dense, task::clustering)
-INSTANTIATE_NON_DISTRIBUTED(float, method::lloyd_csr, task::clustering)
-INSTANTIATE_NON_DISTRIBUTED(double, method::lloyd_csr, task::clustering)
+INSTANTIATE(float, method::lloyd_csr, task::clustering)
+INSTANTIATE(double, method::lloyd_csr, task::clustering)
 
 } // namespace v1
 } // namespace oneapi::dal::kmeans::detail