Make sure metrics work with column-wise distributed training (#9020)

2023-04-17 12:48:23 -07:00
parent 191d0aa5cf
commit ba9d24ff7b
16 changed files with 1183 additions and 811 deletions
--- a/src/metric/survival_metric.cu
+++ b/src/metric/survival_metric.cu
@@ -212,7 +212,9 @@ struct EvalEWiseSurvivalBase : public MetricNoCache {
                                  info.labels_upper_bound_, preds);

    double dat[2]{result.Residue(), result.Weights()};
-    collective::Allreduce<collective::Operation::kSum>(dat, 2);
+    if (info.IsRowSplit()) {
+      collective::Allreduce<collective::Operation::kSum>(dat, 2);
+    }
    return Policy::GetFinal(dat[0], dat[1]);
  }