Update cpp/mrc/include/mrc/node/rx_node.hpp

Co-authored-by: Michael Demoret <42954918+mdemoret-nv@users.noreply.github.com>
nv-morpheus · May 9, 2023 · c13d2ba · c13d2ba
1 parent 892fdf8
commit c13d2ba
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cpp/mrc/include/mrc/node/rx_node.hpp b/cpp/mrc/include/mrc/node/rx_node.hpp
@@ -235,9 +235,11 @@ class RxNodeComponent : public WritableProvider<InputT>, public WritableAcceptor
                 this->get_writable_edge()->await_write(std::move(message));
             },
             [this](std::exception_ptr ptr) {
+                runnable::Context::get_runtime_context().set_exception(std::move(ptr));
+            },
+            [this]() {
                 // On completion, release connections
                 WritableAcceptor<OutputT>::release_edge_connection();
-                runnable::Context::get_runtime_context().set_exception(std::move(ptr));
             }));
     }