Distributed K-Means: fix for step #2

amgrigoriev · Dec 9, 2021 · ff495f3 · ff495f3
1 parent c5d36a9
commit ff495f3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cpp/daal/src/algorithms/kmeans/oneapi/cl_kernels/kmeans_cl_kernels_distr_steps.cl b/cpp/daal/src/algorithms/kmeans/oneapi/cl_kernels/kmeans_cl_kernels_distr_steps.cl
@@ -45,9 +45,9 @@ DECLARE_SOURCE(
         const int local_id                  = get_local_id(1);
         const int oldN                      = partialCentroidsCounters[global_id];
         const int newN                      = cCounters[global_id];
-        const algorithmFPType oldContrib    = oldN * partialCentroids[global_id * P + local_id];
-        const algorithmFPType newContrib    = newN * centroids[global_id * P + local_id];
-        centroids[global_id * P + local_id] = (oldContrib + newContrib) / (oldN + newN);
+        const algorithmFPType oldContrib    = oldN > 0 ? oldN * partialCentroids[global_id * P + local_id] : 0.0;
+        const algorithmFPType newContrib    = newN > 0 ? newN * centroids[global_id * P + local_id] : 0.0;
+        centroids[global_id * P + local_id] = (oldN + newN > 0) ? (oldContrib + newContrib) / (oldN + newN) : 0.0;
         if (local_id == 0) cCounters[global_id] = oldN + newN;
     }