error check

rapidsai · Feb 17, 2021 · 30e341f · 30e341f
1 parent def166f
commit 30e341f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cpp/include/raft/handle.hpp b/cpp/include/raft/handle.hpp
@@ -81,6 +81,9 @@ class handle_t {
   handle_t(const handle_t& other, int stream_id,
            int n_streams = kNumDefaultWorkerStreams)
     : dev_id_(other.get_device()), streams_(n_streams) {
+    RAFT_EXPECTS(
+      other.get_num_internal_streams() > 0,
+      "ERROR: the main handle must have at least one worker stream\n");
     prop_ = other.get_device_properties();
     device_prop_initialized_ = true;
     device_allocator_ = other.get_device_allocator();