public class nccl extends nccl
Modifier and Type | Field and Description |
---|---|
static int |
NCCL_MAJOR
Copyright (c) 2015-2021, NVIDIA CORPORATION.
|
static int |
NCCL_MINOR |
static int |
NCCL_PATCH |
static int |
NCCL_SPLIT_NOCOLOR |
static String |
NCCL_SUFFIX |
static int |
NCCL_UNIQUE_ID_BYTES |
static int |
NCCL_VERSION_CODE |
static int |
ncclAvg
enum ncclRedOp_t
|
static int |
ncclBfloat16
enum ncclDataType_t
|
static int |
ncclChar
enum ncclDataType_t
|
static int |
ncclDouble
enum ncclDataType_t
|
static int |
ncclFloat
enum ncclDataType_t
|
static int |
ncclFloat16
enum ncclDataType_t
|
static int |
ncclFloat32
enum ncclDataType_t
|
static int |
ncclFloat64
enum ncclDataType_t
|
static int |
ncclHalf
enum ncclDataType_t
|
static int |
ncclInProgress
enum ncclResult_t
|
static int |
ncclInt
enum ncclDataType_t
|
static int |
ncclInt32
enum ncclDataType_t
|
static int |
ncclInt64
enum ncclDataType_t
|
static int |
ncclInt8
enum ncclDataType_t
|
static int |
ncclInternalError
enum ncclResult_t
|
static int |
ncclInvalidArgument
enum ncclResult_t
|
static int |
ncclInvalidUsage
enum ncclResult_t
|
static int |
ncclMax
enum ncclRedOp_t
|
static int |
ncclMaxRedOp |
static int |
ncclMin
enum ncclRedOp_t
|
static int |
ncclNumOps
enum ncclRedOp_t
|
static int |
ncclNumOps_dummy
enum ncclRedOp_dummy_t
|
static int |
ncclNumResults
enum ncclResult_t
|
static int |
ncclNumTypes
enum ncclDataType_t
|
static int |
ncclProd
enum ncclRedOp_t
|
static int |
ncclRemoteError
enum ncclResult_t
|
static int |
ncclScalarDevice
enum ncclScalarResidence_t
|
static int |
ncclScalarHostImmediate
enum ncclScalarResidence_t
|
static int |
ncclSuccess
enum ncclResult_t
|
static int |
ncclSum
enum ncclRedOp_t
|
static int |
ncclSystemError
enum ncclResult_t
|
static int |
ncclUint32
enum ncclDataType_t
|
static int |
ncclUint64
enum ncclDataType_t
|
static int |
ncclUint8
enum ncclDataType_t
|
static int |
ncclUnhandledCudaError
enum ncclResult_t
|
Constructor and Description |
---|
nccl() |
Modifier and Type | Method and Description |
---|---|
static int |
ncclAllGather(Pointer sendbuff,
Pointer recvbuff,
long sendcount,
int datatype,
ncclComm comm,
CUstream_st stream) |
static int |
ncclAllReduce(Pointer sendbuff,
Pointer recvbuff,
long count,
int datatype,
int op,
ncclComm comm,
CUstream_st stream) |
static int |
ncclBcast(Pointer buff,
long count,
int datatype,
int root,
ncclComm comm,
CUstream_st stream) |
static int |
ncclBroadcast(Pointer sendbuff,
Pointer recvbuff,
long count,
int datatype,
int root,
ncclComm comm,
CUstream_st stream) |
static int |
ncclCommAbort(ncclComm comm) |
static int |
ncclCommCount(ncclComm comm,
int[] count) |
static int |
ncclCommCount(ncclComm comm,
IntBuffer count) |
static int |
ncclCommCount(ncclComm comm,
IntPointer count) |
static int |
ncclCommCuDevice(ncclComm comm,
int[] device) |
static int |
ncclCommCuDevice(ncclComm comm,
IntBuffer device) |
static int |
ncclCommCuDevice(ncclComm comm,
IntPointer device) |
static int |
ncclCommDeregister(ncclComm comm,
Pointer handle) |
static int |
ncclCommDestroy(ncclComm comm) |
static int |
ncclCommFinalize(ncclComm comm) |
static int |
ncclCommGetAsyncError(ncclComm comm,
int[] asyncError) |
static int |
ncclCommGetAsyncError(ncclComm comm,
IntBuffer asyncError) |
static int |
ncclCommGetAsyncError(ncclComm comm,
IntPointer asyncError) |
static int |
ncclCommInitAll(ncclComm comm,
int ndev,
int[] devlist) |
static int |
ncclCommInitAll(ncclComm comm,
int ndev,
IntBuffer devlist) |
static int |
ncclCommInitAll(ncclComm comm,
int ndev,
IntPointer devlist) |
static int |
ncclCommInitAll(PointerPointer comm,
int ndev,
int[] devlist) |
static int |
ncclCommInitAll(PointerPointer comm,
int ndev,
IntBuffer devlist) |
static int |
ncclCommInitAll(PointerPointer comm,
int ndev,
IntPointer devlist) |
static int |
ncclCommInitRank(ncclComm comm,
int nranks,
ncclUniqueId commId,
int rank) |
static int |
ncclCommInitRank(PointerPointer comm,
int nranks,
ncclUniqueId commId,
int rank) |
static int |
ncclCommInitRankConfig(ncclComm comm,
int nranks,
ncclUniqueId commId,
int rank,
ncclConfig_t config) |
static int |
ncclCommInitRankConfig(PointerPointer comm,
int nranks,
ncclUniqueId commId,
int rank,
ncclConfig_t config) |
static int |
ncclCommRegister(ncclComm comm,
Pointer buff,
long size,
Pointer handle) |
static int |
ncclCommRegister(ncclComm comm,
Pointer buff,
long size,
PointerPointer handle) |
static int |
ncclCommSplit(ncclComm comm,
int color,
int key,
ncclComm newcomm,
ncclConfig_t config) |
static int |
ncclCommSplit(ncclComm comm,
int color,
int key,
PointerPointer newcomm,
ncclConfig_t config) |
static int |
ncclCommUserRank(ncclComm comm,
int[] rank) |
static int |
ncclCommUserRank(ncclComm comm,
IntBuffer rank) |
static int |
ncclCommUserRank(ncclComm comm,
IntPointer rank) |
static BytePointer |
ncclGetErrorString(int result) |
static BytePointer |
ncclGetLastError(ncclComm comm) |
static int |
ncclGetUniqueId(ncclUniqueId uniqueId) |
static int |
ncclGetVersion(int[] version) |
static int |
ncclGetVersion(IntBuffer version) |
static int |
ncclGetVersion(IntPointer version) |
static int |
ncclGroupEnd() |
static int |
ncclGroupStart() |
static int |
ncclMaxRedOp() |
static int |
ncclMemAlloc(Pointer ptr,
long size) |
static int |
ncclMemAlloc(PointerPointer ptr,
long size) |
static int |
ncclMemFree(Pointer ptr) |
static int |
ncclRecv(Pointer recvbuff,
long count,
int datatype,
int peer,
ncclComm comm,
CUstream_st stream) |
static int |
ncclRedOpCreatePreMulSum(int[] op,
Pointer scalar,
int datatype,
int residence,
ncclComm comm) |
static int |
ncclRedOpCreatePreMulSum(IntBuffer op,
Pointer scalar,
int datatype,
int residence,
ncclComm comm) |
static int |
ncclRedOpCreatePreMulSum(IntPointer op,
Pointer scalar,
int datatype,
int residence,
ncclComm comm) |
static int |
ncclRedOpDestroy(int op,
ncclComm comm) |
static int |
ncclReduce(Pointer sendbuff,
Pointer recvbuff,
long count,
int datatype,
int op,
int root,
ncclComm comm,
CUstream_st stream) |
static int |
ncclReduceScatter(Pointer sendbuff,
Pointer recvbuff,
long recvcount,
int datatype,
int op,
ncclComm comm,
CUstream_st stream) |
static int |
ncclSend(Pointer sendbuff,
long count,
int datatype,
int peer,
ncclComm comm,
CUstream_st stream) |
static int |
pncclAllGather(Pointer sendbuff,
Pointer recvbuff,
long sendcount,
int datatype,
ncclComm comm,
CUstream_st stream) |
static int |
pncclAllReduce(Pointer sendbuff,
Pointer recvbuff,
long count,
int datatype,
int op,
ncclComm comm,
CUstream_st stream) |
static int |
pncclBcast(Pointer buff,
long count,
int datatype,
int root,
ncclComm comm,
CUstream_st stream) |
static int |
pncclBroadcast(Pointer sendbuff,
Pointer recvbuff,
long count,
int datatype,
int root,
ncclComm comm,
CUstream_st stream) |
static int |
pncclCommAbort(ncclComm comm) |
static int |
pncclCommCount(ncclComm comm,
int[] count) |
static int |
pncclCommCount(ncclComm comm,
IntBuffer count) |
static int |
pncclCommCount(ncclComm comm,
IntPointer count) |
static int |
pncclCommCuDevice(ncclComm comm,
int[] device) |
static int |
pncclCommCuDevice(ncclComm comm,
IntBuffer device) |
static int |
pncclCommCuDevice(ncclComm comm,
IntPointer device) |
static int |
pncclCommDeregister(ncclComm comm,
Pointer handle) |
static int |
pncclCommDestroy(ncclComm comm) |
static int |
pncclCommFinalize(ncclComm comm) |
static int |
pncclCommGetAsyncError(ncclComm comm,
int[] asyncError) |
static int |
pncclCommGetAsyncError(ncclComm comm,
IntBuffer asyncError) |
static int |
pncclCommGetAsyncError(ncclComm comm,
IntPointer asyncError) |
static int |
pncclCommInitAll(ncclComm comm,
int ndev,
int[] devlist) |
static int |
pncclCommInitAll(ncclComm comm,
int ndev,
IntBuffer devlist) |
static int |
pncclCommInitAll(ncclComm comm,
int ndev,
IntPointer devlist) |
static int |
pncclCommInitAll(PointerPointer comm,
int ndev,
int[] devlist) |
static int |
pncclCommInitAll(PointerPointer comm,
int ndev,
IntBuffer devlist) |
static int |
pncclCommInitAll(PointerPointer comm,
int ndev,
IntPointer devlist) |
static int |
pncclCommInitRank(ncclComm comm,
int nranks,
ncclUniqueId commId,
int rank) |
static int |
pncclCommInitRank(PointerPointer comm,
int nranks,
ncclUniqueId commId,
int rank) |
static int |
pncclCommInitRankConfig(ncclComm comm,
int nranks,
ncclUniqueId commId,
int rank,
ncclConfig_t config) |
static int |
pncclCommInitRankConfig(PointerPointer comm,
int nranks,
ncclUniqueId commId,
int rank,
ncclConfig_t config) |
static int |
pncclCommRegister(ncclComm comm,
Pointer buff,
long size,
Pointer handle) |
static int |
pncclCommRegister(ncclComm comm,
Pointer buff,
long size,
PointerPointer handle) |
static int |
pncclCommSplit(ncclComm comm,
int color,
int key,
ncclComm newcomm,
ncclConfig_t config) |
static int |
pncclCommSplit(ncclComm comm,
int color,
int key,
PointerPointer newcomm,
ncclConfig_t config) |
static int |
pncclCommUserRank(ncclComm comm,
int[] rank) |
static int |
pncclCommUserRank(ncclComm comm,
IntBuffer rank) |
static int |
pncclCommUserRank(ncclComm comm,
IntPointer rank) |
static BytePointer |
pncclGetErrorString(int result) |
static BytePointer |
pncclGetLastError(ncclComm comm) |
static int |
pncclGetUniqueId(ncclUniqueId uniqueId) |
static int |
pncclGetVersion(int[] version) |
static int |
pncclGetVersion(IntBuffer version) |
static int |
pncclGetVersion(IntPointer version) |
static int |
pncclGroupEnd() |
static int |
pncclGroupStart() |
static int |
pncclMemAlloc(Pointer ptr,
long size) |
static int |
pncclMemAlloc(PointerPointer ptr,
long size) |
static int |
pncclMemFree(Pointer ptr) |
static int |
pncclRecv(Pointer recvbuff,
long count,
int datatype,
int peer,
ncclComm comm,
CUstream_st stream) |
static int |
pncclRedOpCreatePreMulSum(int[] op,
Pointer scalar,
int datatype,
int residence,
ncclComm comm) |
static int |
pncclRedOpCreatePreMulSum(IntBuffer op,
Pointer scalar,
int datatype,
int residence,
ncclComm comm) |
static int |
pncclRedOpCreatePreMulSum(IntPointer op,
Pointer scalar,
int datatype,
int residence,
ncclComm comm) |
static int |
pncclRedOpDestroy(int op,
ncclComm comm) |
static int |
pncclReduce(Pointer sendbuff,
Pointer recvbuff,
long count,
int datatype,
int op,
int root,
ncclComm comm,
CUstream_st stream) |
static int |
pncclReduceScatter(Pointer sendbuff,
Pointer recvbuff,
long recvcount,
int datatype,
int op,
ncclComm comm,
CUstream_st stream) |
static int |
pncclSend(Pointer sendbuff,
long count,
int datatype,
int peer,
ncclComm comm,
CUstream_st stream) |
public static final int NCCL_MAJOR
public static final int NCCL_MINOR
public static final int NCCL_PATCH
public static final String NCCL_SUFFIX
public static final int NCCL_VERSION_CODE
public static final int NCCL_UNIQUE_ID_BYTES
public static final int ncclSuccess
public static final int ncclUnhandledCudaError
public static final int ncclSystemError
public static final int ncclInternalError
public static final int ncclInvalidArgument
public static final int ncclInvalidUsage
public static final int ncclRemoteError
public static final int ncclInProgress
public static final int ncclNumResults
public static final int NCCL_SPLIT_NOCOLOR
public static final int ncclNumOps_dummy
public static final int ncclSum
public static final int ncclProd
public static final int ncclMax
public static final int ncclMin
public static final int ncclAvg
public static final int ncclNumOps
public static final int ncclMaxRedOp
public static final int ncclInt8
public static final int ncclChar
public static final int ncclUint8
public static final int ncclInt32
public static final int ncclInt
public static final int ncclUint32
public static final int ncclInt64
public static final int ncclUint64
public static final int ncclFloat16
public static final int ncclHalf
public static final int ncclFloat32
public static final int ncclFloat
public static final int ncclFloat64
public static final int ncclDouble
public static final int ncclBfloat16
public static final int ncclNumTypes
public static final int ncclScalarDevice
public static final int ncclScalarHostImmediate
@Cast(value="ncclResult_t") public static int ncclMemAlloc(@Cast(value="void**") PointerPointer ptr, @Cast(value="size_t") long size)
@Cast(value="ncclResult_t") public static int ncclMemAlloc(@Cast(value="void**") @ByPtrPtr Pointer ptr, @Cast(value="size_t") long size)
@Cast(value="ncclResult_t") public static int pncclMemAlloc(@Cast(value="void**") PointerPointer ptr, @Cast(value="size_t") long size)
@Cast(value="ncclResult_t") public static int pncclMemAlloc(@Cast(value="void**") @ByPtrPtr Pointer ptr, @Cast(value="size_t") long size)
@Cast(value="ncclResult_t") public static int ncclGetVersion(IntPointer version)
@Cast(value="ncclResult_t") public static int pncclGetVersion(IntPointer version)
@Cast(value="ncclResult_t") public static int ncclGetUniqueId(ncclUniqueId uniqueId)
@Cast(value="ncclResult_t") public static int pncclGetUniqueId(ncclUniqueId uniqueId)
@Cast(value="ncclResult_t") public static int ncclCommInitRankConfig(@ByPtrPtr ncclComm comm, int nranks, @ByVal ncclUniqueId commId, int rank, ncclConfig_t config)
@Cast(value="ncclResult_t") public static int ncclCommInitRankConfig(@Cast(value="ncclComm**") PointerPointer comm, int nranks, @ByVal ncclUniqueId commId, int rank, ncclConfig_t config)
@Cast(value="ncclResult_t") public static int pncclCommInitRankConfig(@ByPtrPtr ncclComm comm, int nranks, @ByVal ncclUniqueId commId, int rank, ncclConfig_t config)
@Cast(value="ncclResult_t") public static int pncclCommInitRankConfig(@Cast(value="ncclComm**") PointerPointer comm, int nranks, @ByVal ncclUniqueId commId, int rank, ncclConfig_t config)
@Cast(value="ncclResult_t") public static int ncclCommInitRank(@ByPtrPtr ncclComm comm, int nranks, @ByVal ncclUniqueId commId, int rank)
@Cast(value="ncclResult_t") public static int ncclCommInitRank(@Cast(value="ncclComm**") PointerPointer comm, int nranks, @ByVal ncclUniqueId commId, int rank)
@Cast(value="ncclResult_t") public static int pncclCommInitRank(@ByPtrPtr ncclComm comm, int nranks, @ByVal ncclUniqueId commId, int rank)
@Cast(value="ncclResult_t") public static int pncclCommInitRank(@Cast(value="ncclComm**") PointerPointer comm, int nranks, @ByVal ncclUniqueId commId, int rank)
@Cast(value="ncclResult_t") public static int ncclCommInitAll(@ByPtrPtr ncclComm comm, int ndev, @Const IntPointer devlist)
@Cast(value="ncclResult_t") public static int ncclCommInitAll(@Cast(value="ncclComm**") PointerPointer comm, int ndev, @Const IntBuffer devlist)
@Cast(value="ncclResult_t") public static int ncclCommInitAll(@ByPtrPtr ncclComm comm, int ndev, @Const int[] devlist)
@Cast(value="ncclResult_t") public static int ncclCommInitAll(@Cast(value="ncclComm**") PointerPointer comm, int ndev, @Const IntPointer devlist)
@Cast(value="ncclResult_t") public static int ncclCommInitAll(@ByPtrPtr ncclComm comm, int ndev, @Const IntBuffer devlist)
@Cast(value="ncclResult_t") public static int ncclCommInitAll(@Cast(value="ncclComm**") PointerPointer comm, int ndev, @Const int[] devlist)
@Cast(value="ncclResult_t") public static int pncclCommInitAll(@ByPtrPtr ncclComm comm, int ndev, @Const IntPointer devlist)
@Cast(value="ncclResult_t") public static int pncclCommInitAll(@Cast(value="ncclComm**") PointerPointer comm, int ndev, @Const IntBuffer devlist)
@Cast(value="ncclResult_t") public static int pncclCommInitAll(@ByPtrPtr ncclComm comm, int ndev, @Const int[] devlist)
@Cast(value="ncclResult_t") public static int pncclCommInitAll(@Cast(value="ncclComm**") PointerPointer comm, int ndev, @Const IntPointer devlist)
@Cast(value="ncclResult_t") public static int pncclCommInitAll(@ByPtrPtr ncclComm comm, int ndev, @Const IntBuffer devlist)
@Cast(value="ncclResult_t") public static int pncclCommInitAll(@Cast(value="ncclComm**") PointerPointer comm, int ndev, @Const int[] devlist)
@Cast(value="ncclResult_t") public static int ncclCommSplit(ncclComm comm, int color, int key, @ByPtrPtr ncclComm newcomm, ncclConfig_t config)
@Cast(value="ncclResult_t") public static int ncclCommSplit(ncclComm comm, int color, int key, @Cast(value="ncclComm**") PointerPointer newcomm, ncclConfig_t config)
@Cast(value="ncclResult_t") public static int pncclCommSplit(ncclComm comm, int color, int key, @ByPtrPtr ncclComm newcomm, ncclConfig_t config)
@Cast(value="ncclResult_t") public static int pncclCommSplit(ncclComm comm, int color, int key, @Cast(value="ncclComm**") PointerPointer newcomm, ncclConfig_t config)
@Cast(value="const char*") public static BytePointer ncclGetErrorString(@Cast(value="ncclResult_t") int result)
@Cast(value="const char*") public static BytePointer pncclGetErrorString(@Cast(value="ncclResult_t") int result)
@Cast(value="const char*") public static BytePointer ncclGetLastError(ncclComm comm)
@Cast(value="const char*") public static BytePointer pncclGetLastError(ncclComm comm)
@Cast(value="ncclResult_t") public static int ncclCommGetAsyncError(ncclComm comm, @Cast(value="ncclResult_t*") IntPointer asyncError)
@Cast(value="ncclResult_t") public static int ncclCommGetAsyncError(ncclComm comm, @Cast(value="ncclResult_t*") IntBuffer asyncError)
@Cast(value="ncclResult_t") public static int ncclCommGetAsyncError(ncclComm comm, @Cast(value="ncclResult_t*") int[] asyncError)
@Cast(value="ncclResult_t") public static int pncclCommGetAsyncError(ncclComm comm, @Cast(value="ncclResult_t*") IntPointer asyncError)
@Cast(value="ncclResult_t") public static int pncclCommGetAsyncError(ncclComm comm, @Cast(value="ncclResult_t*") IntBuffer asyncError)
@Cast(value="ncclResult_t") public static int pncclCommGetAsyncError(ncclComm comm, @Cast(value="ncclResult_t*") int[] asyncError)
@Cast(value="ncclResult_t") public static int ncclCommCount(ncclComm comm, IntPointer count)
@Cast(value="ncclResult_t") public static int ncclCommCount(ncclComm comm, IntBuffer count)
@Cast(value="ncclResult_t") public static int ncclCommCount(ncclComm comm, int[] count)
@Cast(value="ncclResult_t") public static int pncclCommCount(ncclComm comm, IntPointer count)
@Cast(value="ncclResult_t") public static int pncclCommCount(ncclComm comm, IntBuffer count)
@Cast(value="ncclResult_t") public static int pncclCommCount(ncclComm comm, int[] count)
@Cast(value="ncclResult_t") public static int ncclCommCuDevice(ncclComm comm, IntPointer device)
@Cast(value="ncclResult_t") public static int ncclCommCuDevice(ncclComm comm, IntBuffer device)
@Cast(value="ncclResult_t") public static int ncclCommCuDevice(ncclComm comm, int[] device)
@Cast(value="ncclResult_t") public static int pncclCommCuDevice(ncclComm comm, IntPointer device)
@Cast(value="ncclResult_t") public static int pncclCommCuDevice(ncclComm comm, IntBuffer device)
@Cast(value="ncclResult_t") public static int pncclCommCuDevice(ncclComm comm, int[] device)
@Cast(value="ncclResult_t") public static int ncclCommUserRank(ncclComm comm, IntPointer rank)
@Cast(value="ncclResult_t") public static int ncclCommUserRank(ncclComm comm, IntBuffer rank)
@Cast(value="ncclResult_t") public static int ncclCommUserRank(ncclComm comm, int[] rank)
@Cast(value="ncclResult_t") public static int pncclCommUserRank(ncclComm comm, IntPointer rank)
@Cast(value="ncclResult_t") public static int pncclCommUserRank(ncclComm comm, IntBuffer rank)
@Cast(value="ncclResult_t") public static int pncclCommUserRank(ncclComm comm, int[] rank)
@MemberGetter public static int ncclMaxRedOp()
@Cast(value="ncclResult_t") public static int ncclRedOpCreatePreMulSum(@Cast(value="ncclRedOp_t*") IntPointer op, Pointer scalar, @Cast(value="ncclDataType_t") int datatype, @Cast(value="ncclScalarResidence_t") int residence, ncclComm comm)
@Cast(value="ncclResult_t") public static int ncclRedOpCreatePreMulSum(@Cast(value="ncclRedOp_t*") IntBuffer op, Pointer scalar, @Cast(value="ncclDataType_t") int datatype, @Cast(value="ncclScalarResidence_t") int residence, ncclComm comm)
@Cast(value="ncclResult_t") public static int ncclRedOpCreatePreMulSum(@Cast(value="ncclRedOp_t*") int[] op, Pointer scalar, @Cast(value="ncclDataType_t") int datatype, @Cast(value="ncclScalarResidence_t") int residence, ncclComm comm)
@Cast(value="ncclResult_t") public static int pncclRedOpCreatePreMulSum(@Cast(value="ncclRedOp_t*") IntPointer op, Pointer scalar, @Cast(value="ncclDataType_t") int datatype, @Cast(value="ncclScalarResidence_t") int residence, ncclComm comm)
@Cast(value="ncclResult_t") public static int pncclRedOpCreatePreMulSum(@Cast(value="ncclRedOp_t*") IntBuffer op, Pointer scalar, @Cast(value="ncclDataType_t") int datatype, @Cast(value="ncclScalarResidence_t") int residence, ncclComm comm)
@Cast(value="ncclResult_t") public static int pncclRedOpCreatePreMulSum(@Cast(value="ncclRedOp_t*") int[] op, Pointer scalar, @Cast(value="ncclDataType_t") int datatype, @Cast(value="ncclScalarResidence_t") int residence, ncclComm comm)
@Cast(value="ncclResult_t") public static int ncclRedOpDestroy(@Cast(value="ncclRedOp_t") int op, ncclComm comm)
@Cast(value="ncclResult_t") public static int pncclRedOpDestroy(@Cast(value="ncclRedOp_t") int op, ncclComm comm)
@Cast(value="ncclResult_t") public static int ncclReduce(@Const Pointer sendbuff, Pointer recvbuff, @Cast(value="size_t") long count, @Cast(value="ncclDataType_t") int datatype, @Cast(value="ncclRedOp_t") int op, int root, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int pncclReduce(@Const Pointer sendbuff, Pointer recvbuff, @Cast(value="size_t") long count, @Cast(value="ncclDataType_t") int datatype, @Cast(value="ncclRedOp_t") int op, int root, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int ncclBcast(Pointer buff, @Cast(value="size_t") long count, @Cast(value="ncclDataType_t") int datatype, int root, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int pncclBcast(Pointer buff, @Cast(value="size_t") long count, @Cast(value="ncclDataType_t") int datatype, int root, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int ncclBroadcast(@Const Pointer sendbuff, Pointer recvbuff, @Cast(value="size_t") long count, @Cast(value="ncclDataType_t") int datatype, int root, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int pncclBroadcast(@Const Pointer sendbuff, Pointer recvbuff, @Cast(value="size_t") long count, @Cast(value="ncclDataType_t") int datatype, int root, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int ncclAllReduce(@Const Pointer sendbuff, Pointer recvbuff, @Cast(value="size_t") long count, @Cast(value="ncclDataType_t") int datatype, @Cast(value="ncclRedOp_t") int op, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int pncclAllReduce(@Const Pointer sendbuff, Pointer recvbuff, @Cast(value="size_t") long count, @Cast(value="ncclDataType_t") int datatype, @Cast(value="ncclRedOp_t") int op, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int ncclReduceScatter(@Const Pointer sendbuff, Pointer recvbuff, @Cast(value="size_t") long recvcount, @Cast(value="ncclDataType_t") int datatype, @Cast(value="ncclRedOp_t") int op, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int pncclReduceScatter(@Const Pointer sendbuff, Pointer recvbuff, @Cast(value="size_t") long recvcount, @Cast(value="ncclDataType_t") int datatype, @Cast(value="ncclRedOp_t") int op, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int ncclAllGather(@Const Pointer sendbuff, Pointer recvbuff, @Cast(value="size_t") long sendcount, @Cast(value="ncclDataType_t") int datatype, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int pncclAllGather(@Const Pointer sendbuff, Pointer recvbuff, @Cast(value="size_t") long sendcount, @Cast(value="ncclDataType_t") int datatype, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int ncclSend(@Const Pointer sendbuff, @Cast(value="size_t") long count, @Cast(value="ncclDataType_t") int datatype, int peer, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int pncclSend(@Const Pointer sendbuff, @Cast(value="size_t") long count, @Cast(value="ncclDataType_t") int datatype, int peer, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int pncclRecv(Pointer recvbuff, @Cast(value="size_t") long count, @Cast(value="ncclDataType_t") int datatype, int peer, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int ncclRecv(Pointer recvbuff, @Cast(value="size_t") long count, @Cast(value="ncclDataType_t") int datatype, int peer, ncclComm comm, CUstream_st stream)
@Cast(value="ncclResult_t") public static int ncclCommRegister(ncclComm comm, Pointer buff, @Cast(value="size_t") long size, @Cast(value="void**") PointerPointer handle)
@Cast(value="ncclResult_t") public static int ncclCommRegister(ncclComm comm, Pointer buff, @Cast(value="size_t") long size, @Cast(value="void**") @ByPtrPtr Pointer handle)
@Cast(value="ncclResult_t") public static int pncclCommRegister(ncclComm comm, Pointer buff, @Cast(value="size_t") long size, @Cast(value="void**") PointerPointer handle)
@Cast(value="ncclResult_t") public static int pncclCommRegister(ncclComm comm, Pointer buff, @Cast(value="size_t") long size, @Cast(value="void**") @ByPtrPtr Pointer handle)
@Cast(value="ncclResult_t") public static int ncclCommDeregister(ncclComm comm, Pointer handle)
Copyright © 2024. All rights reserved.