mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-13 01:36:10 +00:00
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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user