Mojo module

ccl

Aliases

`CCL_LIBRARY`

alias CCL_LIBRARY = _Global["CCL_LIBRARY", _init_ccl_dylib]

`CCLAllGatherFn`

alias CCLAllGatherFn = fn(LegacyUnsafePointer[NoneType], LegacyUnsafePointer[NoneType], Int, ncclDataType_t, LegacyUnsafePointer[NoneType], LegacyUnsafePointer[NoneType]) -> ncclResult_t

`CCLAllReduceFn`

alias CCLAllReduceFn = fn(LegacyUnsafePointer[NoneType], LegacyUnsafePointer[NoneType], Int, ncclDataType_t, ncclRedOp_t, LegacyUnsafePointer[NoneType], LegacyUnsafePointer[NoneType]) -> ncclResult_t

`NCCL_LIBRARY_PATHS`

alias NCCL_LIBRARY_PATHS = List[Path]("libnccl.so", "libnccl.so.2", "/usr/lib/x86_64-linux-gnu/libnccl.so", "/usr/lib/x86_64-linux-gnu/libnccl.so.2", Tuple[]())

`ncclComm_t`

alias ncclComm_t = LegacyOpaquePointer

`RCCL_LIBRARY_PATHS`

alias RCCL_LIBRARY_PATHS = List[Path]("librccl.so", "librccl.so.1", "/opt/rocm/lib/librccl.so", "/opt/rocm/lib/librccl.so.1", Tuple[]())

Structs

Communicators:
ncclDataType_t:
ncclRedOp_t:
ncclResult_t:

Functions

allgather:
allreduce:
is_allgather_available:
is_allreduce_available:
ncclCommInitAll:
ncclGroupEnd:
ncclGroupStart:

Aliases​

CCL_LIBRARY​

CCLAllGatherFn​

CCLAllReduceFn​

NCCL_LIBRARY_PATHS​

ncclComm_t​

RCCL_LIBRARY_PATHS​

Structs​

Functions​