Initial support for column-split cpu predictor (#8676)

2023-01-17 14:33:13 -08:00
parent 980233e648
commit 78396f8a6e
4 changed files with 334 additions and 11 deletions
--- a/src/collective/communicator.h
+++ b/src/collective/communicator.h
@@ -207,6 +207,8 @@ class Communicator {
      result = CommunicatorType::kRabit;
    } else if (!CompareStringsCaseInsensitive("federated", str)) {
      result = CommunicatorType::kFederated;
+    } else if (!CompareStringsCaseInsensitive("in-memory", str)) {
+      result = CommunicatorType::kInMemory;
    } else {
      LOG(FATAL) << "Unknown communicator type " << str;
    }