basic init test

2026-05-13 17:55:52 +00:00 · 2023-03-24 23:45:29 +00:00
parent 8b6e35d5e0
commit 3b1abaaad1
2 changed files with 48 additions and 8 deletions
--- a/python/src/_py_mscclpp.cpp
+++ b/python/src/_py_mscclpp.cpp
@@ -247,5 +247,16 @@ NB_MODULE(_py_mscclpp, m) {
      .def(
          "__del__",
          &MscclppComm::close,
+          nb::call_guard<nb::gil_scoped_release>())
+      .def(
+          "bootstrap_all_gather",
+          [](MscclppComm &comm, void *data, int size) {
+            comm.check_open();
+            return maybe(
+                mscclppBootstrapAllGather(comm._handle, data, size),
+                true,
+                "Failed to stop MSCCLPP proxy");
+	  },
          nb::call_guard<nb::gil_scoped_release>());
+
 }