Support CudaIpc connection within a single process (#593)

* Allow CudaIpc connection between GPUs in a single process
* Added an example of connection in a single process
* Minor interface updates

---------

Co-authored-by: Binyang Li <binyli@microsoft.com>
This commit is contained in:
Changho Hwang
2025-08-02 12:59:36 +08:00
committed by GitHub
parent c3b47c59fd
commit c580e4c503
13 changed files with 262 additions and 49 deletions

View File

@@ -77,7 +77,6 @@ void register_core(nb::module_& m) {
nb::enum_<Transport>(m, "Transport")
.value("Unknown", Transport::Unknown)
.value("CudaIpc", Transport::CudaIpc)
.value("Nvls", Transport::Nvls)
.value("IB0", Transport::IB0)
.value("IB1", Transport::IB1)
.value("IB2", Transport::IB2)