Fix alltoallv build and Python import errors

- Add missing accumDtype and symmetricMemory params to NativeAlgorithm
  lambda signatures in alltoallv_fullmesh.cu to match KernelFunc and
  ContextKeyGenFunc typedefs
- Use Cpp-prefixed binding names in alltoallv_single.py imports
  (CppCommunicator, CppDataType, etc.)
- Add missing symmetric_memory=False arg to algo.execute() call
- Fix test imports to use mscclpp public API instead of raw _mscclpp
This commit is contained in:
Qinghua Zhou
2026-04-13 22:43:42 +00:00
parent 215118ae39
commit eb1e0f33ee
3 changed files with 12 additions and 9 deletions

View File

@@ -17,11 +17,11 @@ import torch
import torch.distributed as dist
from typing import Optional, List, Tuple
from mscclpp._mscclpp import (
Communicator,
TcpBootstrap,
DataType,
ReduceOp,
CommResult,
CppCommunicator as Communicator,
CppTcpBootstrap as TcpBootstrap,
CppDataType as DataType,
CppReduceOp as ReduceOp,
CppCommResult as CommResult,
)
from mscclpp.ext.algorithm_collection_builder import AlgorithmCollectionBuilder
@@ -323,6 +323,7 @@ class MscclppAlltoAllV:
None, # executor (not needed for native algos)
0, # nblocks (auto)
0, # nthreads_per_block (auto)
False, # symmetric_memory
self._extras,
)

View File

@@ -130,11 +130,11 @@ def main():
print("=" * 60)
# Import after torch.distributed init
from mscclpp._mscclpp import (
from mscclpp import (
Communicator,
TcpBootstrap,
UniqueId,
)
from mscclpp._mscclpp import CppUniqueId as UniqueId
from mscclpp.ext.alltoallv_single import MscclppAlltoAllV
# Create mscclpp communicator with TcpBootstrap