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

@@ -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