[GPU-Plugin] Multi-GPU for grow_gpu_hist histogram method using NVIDIA NCCL. (#2395)

2017-06-11 13:06:08 -04:00
parent e24f25e0c6
commit 41efe32aa5
19 changed files with 2009 additions and 682 deletions
--- a/plugin/updater_gpu/src/updater_gpu.cu
+++ b/plugin/updater_gpu/src/updater_gpu.cu
@@ -1,5 +1,5 @@
 /*!
- * Copyright 2016 Rory Mitchell
+ * Copyright 2017 XGBoost contributors
 */
 #include <xgboost/tree_updater.h>
 #include <vector>
@@ -76,7 +76,7 @@ class GPUHistMaker : public TreeUpdater {
  }

  bool UpdatePredictionCache(const DMatrix* data,
-    std::vector<bst_float>* out_preds) override {
+                             std::vector<bst_float>* out_preds) override {
    return builder.UpdatePredictionCache(data, out_preds);
  }