mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-24 14:54:51 +00:00
Fix Python bindings and tests (#690)
Minimal fix to make things work. We need a more careful look at preventing silent fallback of nanobind when it fails to (properly) construct a C++ STL object with mscclpp instances.
This commit is contained in:
@@ -453,7 +453,10 @@ class MscclppAllReduce6:
|
||||
)
|
||||
|
||||
# create a memory_channel for each remote neighbor
|
||||
self.semaphores = group.make_semaphore(self.nvlink_connections, MemoryDevice2DeviceSemaphore)
|
||||
self.semaphores = {
|
||||
rank: MemoryDevice2DeviceSemaphore(sema)
|
||||
for rank, sema in group.make_semaphores(self.nvlink_connections).items()
|
||||
}
|
||||
file_dir = os.path.dirname(os.path.abspath(__file__))
|
||||
self.kernel = KernelBuilder(
|
||||
file="allreduce.cu",
|
||||
|
||||
Reference in New Issue
Block a user