fix incorrect reduction when there's fewer bins than jobs (#2984)

uxlfoundation · Nov 21, 2024 · 49c0200 · 49c0200
1 parent 14f4e79
commit 49c0200
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cpp/oneapi/dal/algo/decision_forest/backend/gpu/train_helpers.hpp b/cpp/oneapi/dal/algo/decision_forest/backend/gpu/train_helpers.hpp
@@ -39,6 +39,12 @@ inline void merge_stat(Float& dst_count,
                        Float src_s2c) {
     if (Float(0) == src_count)
         return;
+    if (!dst_count) {
+        dst_count = src_count;
+        dst_mean = src_mean;
+        dst_s2c = src_s2c;
+        return;
+    }
 
     Float sum_n1n2 = dst_count + src_count;
     Float mul_n1n2 = dst_count * src_count;