[bug] fix policy changes for spmd (#2554)

* Update dal.cpp

* Update policy.cpp

Author: icfaust

onedal/common/policy.cpp

@@ -147,13 +147,14 @@ py::object get_policy(py::object obj) {
 };
 
 ONEDAL_PY_INIT_MODULE(policy) {
+#ifdef ONEDAL_DATA_PARALLEL_SPMD
+    instantiate_spmd_policy(m);
+#else
     instantiate_host_policy(m);
     instantiate_default_host_policy(m);
 #ifdef ONEDAL_DATA_PARALLEL
     instantiate_data_parallel_policy(m);
 #endif // ONEDAL_DATA_PARALLEL
-#ifdef ONEDAL_DATA_PARALLEL_SPMD
-    instantiate_spmd_policy(m);
 #endif // ONEDAL_DATA_PARALLEL_SPMD
     m.def("get_policy", &get_policy, py::arg("queue") = py::none());
 }

onedal/dal.cpp

@@ -23,6 +23,7 @@ namespace oneapi::dal::python {
 
 /* common */
 #ifdef ONEDAL_DATA_PARALLEL_SPMD
+ONEDAL_PY_INIT_MODULE(policy);
 /* algorithms */
 ONEDAL_PY_INIT_MODULE(covariance);
 ONEDAL_PY_INIT_MODULE(dbscan);
@@ -83,6 +84,7 @@ ONEDAL_PY_INIT_MODULE(finiteness_checker);
 
 #ifdef ONEDAL_DATA_PARALLEL_SPMD
 PYBIND11_MODULE(_onedal_py_spmd_dpc, m) {
+    init_policy(m);
     init_covariance(m);
     init_dbscan(m);
     init_decomposition(m);