public class nccl extends nccl
| Modifier and Type | Field and Description |
|---|---|
static int |
NCCL_MAJOR
Copyright (c) 2015-2020, NVIDIA CORPORATION.
|
static int |
NCCL_MINOR |
static int |
NCCL_PATCH |
static String |
NCCL_SUFFIX |
static int |
NCCL_UNIQUE_ID_BYTES |
static int |
NCCL_VERSION_CODE |
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 |
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 |
ncclMin
enum ncclRedOp_t
|
static int |
ncclNumOps
enum ncclRedOp_t
|
static int |
ncclNumResults
enum ncclResult_t
|
static int |
ncclNumTypes
enum ncclDataType_t
|
static int |
ncclProd
enum ncclRedOp_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 |
ncclCommDestroy(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 |
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 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 |
ncclRecv(Pointer recvbuff,
long count,
int datatype,
int peer,
ncclComm comm,
CUstream_st stream) |
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 |
pncclCommDestroy(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 |
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 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 |
pncclRecv(Pointer recvbuff,
long count,
int datatype,
int peer,
ncclComm comm,
CUstream_st stream) |
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 ncclNumResults
public static final int ncclSum
public static final int ncclProd
public static final int ncclMax
public static final int ncclMin
public static final int ncclNumOps
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 ncclNumTypes
@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 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="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="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)
@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)
Copyright © 2020. All rights reserved.