rocprofiler-sdk/rccl/details/api_trace.h File Reference

rocprofiler-sdk/rccl/details/api_trace.h File Reference#

Rocprofiler SDK Developer API: rocprofiler-sdk/rccl/details/api_trace.h File Reference

Rocprofiler SDK Developer API 0.5.0

ROCm Profiling API and tools

#include <rocprofiler-sdk/rccl/details/rccl.h>
#include <stddef.h>
#include <stdint.h>

Include dependency graph for api_trace.h:

Data Structures
struct	rcclApiFuncTable

Macros
#define	ROCPROFILER_SDK_USE_SYSTEM_RCCL 0

#define	RCCL_API_TRACE_VERSION_MAJOR 0

#define	RCCL_API_TRACE_VERSION_PATCH 0

#define	RCCL_EXTERN_C_INIT

#define	RCCL_EXTERN_C_FINI

Typedefs
typedef uint64_t	rccl_range_id_t

typedef ncclResult_t(*	ncclAllGather_fn_t) (const void sendbuff, void recvbuff, unsigned long sendcount, ncclDataType_t datatype, ncclComm_t comm, hipStream_t stream)

typedef ncclResult_t(*	ncclAllReduce_fn_t) (const void sendbuff, void recvbuff, unsigned long count, ncclDataType_t datatype, ncclRedOp_t op, struct ncclComm *comm, hipStream_t stream)

typedef ncclResult_t(*	ncclAllToAll_fn_t) (const void sendbuff, void recvbuff, unsigned long count, ncclDataType_t datatype, ncclComm_t comm, hipStream_t stream)

typedef ncclResult_t(*	ncclAllToAllv_fn_t) (const void sendbuff, const unsigned long sendcounts[], const unsigned long sdispls[], void recvbuff, const unsigned long recvcounts[], const unsigned long rdispls[], ncclDataType_t datatype, ncclComm_t comm, hipStream_t stream)

typedef ncclResult_t(*	ncclBroadcast_fn_t) (const void sendbuff, void recvbuff, unsigned long count, ncclDataType_t datatype, int root, ncclComm_t comm, hipStream_t stream)

typedef ncclResult_t(*	ncclGather_fn_t) (const void sendbuff, void recvbuff, unsigned long sendcount, ncclDataType_t datatype, int root, ncclComm_t comm, hipStream_t stream)

typedef ncclResult_t(*	ncclReduce_fn_t) (const void sendbuff, void recvbuff, unsigned long count, ncclDataType_t datatype, ncclRedOp_t op, int root, ncclComm_t comm, hipStream_t stream)

typedef ncclResult_t(*	ncclReduceScatter_fn_t) (const void sendbuff, void recvbuff, unsigned long recvcount, ncclDataType_t datatype, ncclRedOp_t op, struct ncclComm *comm, hipStream_t stream)

typedef ncclResult_t(*	ncclScatter_fn_t) (const void sendbuff, void recvbuff, unsigned long recvcount, ncclDataType_t datatype, int root, ncclComm_t comm, hipStream_t stream)

typedef ncclResult_t(*	ncclSend_fn_t) (const void *sendbuff, unsigned long count, ncclDataType_t datatype, int peer, ncclComm_t comm, hipStream_t stream)

typedef ncclResult_t(*	ncclRecv_fn_t) (void *recvbuff, unsigned long count, ncclDataType_t datatype, int peer, ncclComm_t comm, hipStream_t stream)

typedef ncclResult_t(*	ncclRedOpCreatePreMulSum_fn_t) (ncclRedOp_t op, void scalar, ncclDataType_t datatype, ncclScalarResidence_t residence, ncclComm_t comm)

typedef ncclResult_t(*	ncclRedOpDestroy_fn_t) (ncclRedOp_t op, ncclComm_t comm)

typedef ncclResult_t(*	ncclGroupStart_fn_t) ()

typedef ncclResult_t(*	ncclGroupEnd_fn_t) ()

typedef ncclResult_t(*	ncclGetVersion_fn_t) (int *version)

typedef ncclResult_t(*	ncclGetUniqueId_fn_t) (ncclUniqueId *out)

typedef ncclResult_t(*	ncclCommInitRank_fn_t) (ncclComm_t *newcomm, int nranks, ncclUniqueId commId, int myrank)

typedef ncclResult_t(*	ncclCommInitAll_fn_t) (ncclComm_t comms, int ndev, const int devlist)

typedef ncclResult_t(*	ncclCommInitRankConfig_fn_t) (ncclComm_t comm, int nranks, ncclUniqueId commId, int myrank, ncclConfig_t config)

typedef ncclResult_t(*	ncclCommFinalize_fn_t) (ncclComm_t comm)

typedef ncclResult_t(*	ncclCommDestroy_fn_t) (ncclComm_t comm)

typedef ncclResult_t(*	ncclCommAbort_fn_t) (ncclComm_t comm)

typedef ncclResult_t(*	ncclCommSplit_fn_t) (ncclComm_t comm, int color, int key, ncclComm_t newcomm, ncclConfig_t config)

typedef const char (	ncclGetErrorString_fn_t) (ncclResult_t code)

typedef const char (	ncclGetLastError_fn_t) (const ncclComm_t comm)

typedef ncclResult_t(*	ncclCommGetAsyncError_fn_t) (ncclComm_t comm, ncclResult_t *asyncError)

typedef ncclResult_t(*	ncclCommCount_fn_t) (const ncclComm_t comm, int *count)

typedef ncclResult_t(*	ncclCommCuDevice_fn_t) (const ncclComm_t comm, int *devid)

typedef ncclResult_t(*	ncclCommUserRank_fn_t) (const ncclComm_t comm, int *rank)

typedef ncclResult_t(*	ncclMemAlloc_fn_t) (void **ptr, unsigned long size)

typedef ncclResult_t(*	ncclMemFree_fn_t) (void *ptr)

typedef ncclResult_t(*	mscclLoadAlgo_fn_t) (const char mscclAlgoFilePath, mscclAlgoHandle_t mscclAlgoHandle, int rank)

typedef ncclResult_t(*	mscclRunAlgo_fn_t) (const void sendBuff, const unsigned long sendCounts[], const unsigned long sDisPls[], void recvBuff, const unsigned long recvCounts[], const unsigned long rDisPls[], unsigned long count, ncclDataType_t dataType, int root, int peer, ncclRedOp_t op, mscclAlgoHandle_t mscclAlgoHandle, ncclComm_t comm, hipStream_t stream)

typedef ncclResult_t(*	mscclUnloadAlgo_fn_t) (mscclAlgoHandle_t mscclAlgoHandle)

typedef ncclResult_t(*	ncclCommRegister_fn_t) (const ncclComm_t comm, void buff, unsigned long size, void *handle)

typedef ncclResult_t(*	ncclCommDeregister_fn_t) (const ncclComm_t comm, void *handle)

Data Structure Documentation

◆ rcclApiFuncTable

struct rcclApiFuncTable

Definition at line 227 of file api_trace.h.

Collaboration diagram for rcclApiFuncTable:

Data Fields
mscclLoadAlgo_fn_t	mscclLoadAlgo_fn
mscclRunAlgo_fn_t	mscclRunAlgo_fn
mscclUnloadAlgo_fn_t	mscclUnloadAlgo_fn
ncclAllGather_fn_t	ncclAllGather_fn
ncclAllReduce_fn_t	ncclAllReduce_fn
ncclAllToAll_fn_t	ncclAllToAll_fn
ncclAllToAllv_fn_t	ncclAllToAllv_fn
ncclBroadcast_fn_t	ncclBroadcast_fn
ncclCommAbort_fn_t	ncclCommAbort_fn
ncclCommCount_fn_t	ncclCommCount_fn
ncclCommCuDevice_fn_t	ncclCommCuDevice_fn
ncclCommDeregister_fn_t	ncclCommDeregister_fn
ncclCommDestroy_fn_t	ncclCommDestroy_fn
ncclCommFinalize_fn_t	ncclCommFinalize_fn
ncclCommGetAsyncError_fn_t	ncclCommGetAsyncError_fn
ncclCommInitAll_fn_t	ncclCommInitAll_fn
ncclCommInitRank_fn_t	ncclCommInitRank_fn
ncclCommInitRankConfig_fn_t	ncclCommInitRankConfig_fn
ncclCommRegister_fn_t	ncclCommRegister_fn
ncclCommSplit_fn_t	ncclCommSplit_fn
ncclCommUserRank_fn_t	ncclCommUserRank_fn
ncclGather_fn_t	ncclGather_fn
ncclGetErrorString_fn_t	ncclGetErrorString_fn
ncclGetLastError_fn_t	ncclGetLastError_fn
ncclGetUniqueId_fn_t	ncclGetUniqueId_fn
ncclGetVersion_fn_t	ncclGetVersion_fn
ncclGroupEnd_fn_t	ncclGroupEnd_fn
ncclGroupStart_fn_t	ncclGroupStart_fn
ncclMemAlloc_fn_t	ncclMemAlloc_fn
ncclMemFree_fn_t	ncclMemFree_fn
ncclRecv_fn_t	ncclRecv_fn
ncclRedOpCreatePreMulSum_fn_t	ncclRedOpCreatePreMulSum_fn
ncclRedOpDestroy_fn_t	ncclRedOpDestroy_fn
ncclReduce_fn_t	ncclReduce_fn
ncclReduceScatter_fn_t	ncclReduceScatter_fn
ncclScatter_fn_t	ncclScatter_fn
ncclSend_fn_t	ncclSend_fn
uint64_t	size

Macro Definition Documentation

◆ RCCL_API_TRACE_VERSION_MAJOR

#define RCCL_API_TRACE_VERSION_MAJOR 0

Definition at line 47 of file api_trace.h.

◆ RCCL_API_TRACE_VERSION_PATCH

#define RCCL_API_TRACE_VERSION_PATCH 0

Definition at line 50 of file api_trace.h.

◆ RCCL_EXTERN_C_FINI

#define RCCL_EXTERN_C_FINI

Definition at line 64 of file api_trace.h.

◆ RCCL_EXTERN_C_INIT

#define RCCL_EXTERN_C_INIT

Definition at line 56 of file api_trace.h.

◆ ROCPROFILER_SDK_USE_SYSTEM_RCCL

#define ROCPROFILER_SDK_USE_SYSTEM_RCCL 0

Definition at line 33 of file api_trace.h.

Typedef Documentation

◆ mscclLoadAlgo_fn_t

typedef ncclResult_t(* mscclLoadAlgo_fn_t) (const char *mscclAlgoFilePath, mscclAlgoHandle_t *mscclAlgoHandle, int rank)

Definition at line 199 of file api_trace.h.

◆ mscclRunAlgo_fn_t

typedef ncclResult_t(* mscclRunAlgo_fn_t) (const void *sendBuff, const unsigned long sendCounts[], const unsigned long sDisPls[], void *recvBuff, const unsigned long recvCounts[], const unsigned long rDisPls[], unsigned long count, ncclDataType_t dataType, int root, int peer, ncclRedOp_t op, mscclAlgoHandle_t mscclAlgoHandle, ncclComm_t comm, hipStream_t stream)

Definition at line 203 of file api_trace.h.

◆ mscclUnloadAlgo_fn_t

typedef ncclResult_t(* mscclUnloadAlgo_fn_t) (mscclAlgoHandle_t mscclAlgoHandle)

Definition at line 218 of file api_trace.h.

◆ ncclAllGather_fn_t

typedef ncclResult_t(* ncclAllGather_fn_t) (const void *sendbuff, void *recvbuff, unsigned long sendcount, ncclDataType_t datatype, ncclComm_t comm, hipStream_t stream)

Definition at line 71 of file api_trace.h.

◆ ncclAllReduce_fn_t

typedef ncclResult_t(* ncclAllReduce_fn_t) (const void *sendbuff, void *recvbuff, unsigned long count, ncclDataType_t datatype, ncclRedOp_t op, struct ncclComm *comm, hipStream_t stream)

Definition at line 77 of file api_trace.h.

◆ ncclAllToAll_fn_t

typedef ncclResult_t(* ncclAllToAll_fn_t) (const void *sendbuff, void *recvbuff, unsigned long count, ncclDataType_t datatype, ncclComm_t comm, hipStream_t stream)

Definition at line 84 of file api_trace.h.

◆ ncclAllToAllv_fn_t

typedef ncclResult_t(* ncclAllToAllv_fn_t) (const void *sendbuff, const unsigned long sendcounts[], const unsigned long sdispls[], void *recvbuff, const unsigned long recvcounts[], const unsigned long rdispls[], ncclDataType_t datatype, ncclComm_t comm, hipStream_t stream)

Definition at line 90 of file api_trace.h.

◆ ncclBroadcast_fn_t

typedef ncclResult_t(* ncclBroadcast_fn_t) (const void *sendbuff, void *recvbuff, unsigned long count, ncclDataType_t datatype, int root, ncclComm_t comm, hipStream_t stream)

Definition at line 99 of file api_trace.h.

◆ ncclCommAbort_fn_t

typedef ncclResult_t(* ncclCommAbort_fn_t) (ncclComm_t comm)

Definition at line 175 of file api_trace.h.

◆ ncclCommCount_fn_t

typedef ncclResult_t(* ncclCommCount_fn_t) (const ncclComm_t comm, int *count)

Definition at line 189 of file api_trace.h.

◆ ncclCommCuDevice_fn_t

typedef ncclResult_t(* ncclCommCuDevice_fn_t) (const ncclComm_t comm, int *devid)

Definition at line 191 of file api_trace.h.

◆ ncclCommDeregister_fn_t

typedef ncclResult_t(* ncclCommDeregister_fn_t) (const ncclComm_t comm, void *handle)

Definition at line 225 of file api_trace.h.

◆ ncclCommDestroy_fn_t

typedef ncclResult_t(* ncclCommDestroy_fn_t) (ncclComm_t comm)

Definition at line 173 of file api_trace.h.

◆ ncclCommFinalize_fn_t

typedef ncclResult_t(* ncclCommFinalize_fn_t) (ncclComm_t comm)

Definition at line 171 of file api_trace.h.

◆ ncclCommGetAsyncError_fn_t

typedef ncclResult_t(* ncclCommGetAsyncError_fn_t) (ncclComm_t comm, ncclResult_t *asyncError)

Definition at line 187 of file api_trace.h.

◆ ncclCommInitAll_fn_t

typedef ncclResult_t(* ncclCommInitAll_fn_t) (ncclComm_t *comms, int ndev, const int *devlist)

Definition at line 163 of file api_trace.h.

◆ ncclCommInitRank_fn_t

typedef ncclResult_t(* ncclCommInitRank_fn_t) (ncclComm_t *newcomm, int nranks, ncclUniqueId commId, int myrank)

Definition at line 158 of file api_trace.h.

◆ ncclCommInitRankConfig_fn_t

typedef ncclResult_t(* ncclCommInitRankConfig_fn_t) (ncclComm_t *comm, int nranks, ncclUniqueId commId, int myrank, ncclConfig_t *config)

Definition at line 165 of file api_trace.h.

◆ ncclCommRegister_fn_t

typedef ncclResult_t(* ncclCommRegister_fn_t) (const ncclComm_t comm, void *buff, unsigned long size, void **handle)

Definition at line 220 of file api_trace.h.

◆ ncclCommSplit_fn_t

typedef ncclResult_t(* ncclCommSplit_fn_t) (ncclComm_t comm, int color, int key, ncclComm_t *newcomm, ncclConfig_t *config)

Definition at line 177 of file api_trace.h.

◆ ncclCommUserRank_fn_t

typedef ncclResult_t(* ncclCommUserRank_fn_t) (const ncclComm_t comm, int *rank)

Definition at line 193 of file api_trace.h.

◆ ncclGather_fn_t

typedef ncclResult_t(* ncclGather_fn_t) (const void *sendbuff, void *recvbuff, unsigned long sendcount, ncclDataType_t datatype, int root, ncclComm_t comm, hipStream_t stream)

Definition at line 106 of file api_trace.h.

◆ ncclGetErrorString_fn_t

typedef const char *(* ncclGetErrorString_fn_t) (ncclResult_t code)

Definition at line 183 of file api_trace.h.

◆ ncclGetLastError_fn_t

typedef const char *(* ncclGetLastError_fn_t) (const ncclComm_t comm)

Definition at line 185 of file api_trace.h.

◆ ncclGetUniqueId_fn_t

typedef ncclResult_t(* ncclGetUniqueId_fn_t) (ncclUniqueId *out)

Definition at line 156 of file api_trace.h.

◆ ncclGetVersion_fn_t

typedef ncclResult_t(* ncclGetVersion_fn_t) (int *version)

Definition at line 155 of file api_trace.h.

◆ ncclGroupEnd_fn_t

typedef ncclResult_t(* ncclGroupEnd_fn_t) ()

Definition at line 154 of file api_trace.h.

◆ ncclGroupStart_fn_t

typedef ncclResult_t(* ncclGroupStart_fn_t) ()

Definition at line 153 of file api_trace.h.

◆ ncclMemAlloc_fn_t

typedef ncclResult_t(* ncclMemAlloc_fn_t) (void **ptr, unsigned long size)

Definition at line 195 of file api_trace.h.

◆ ncclMemFree_fn_t

typedef ncclResult_t(* ncclMemFree_fn_t) (void *ptr)

Definition at line 197 of file api_trace.h.

◆ ncclRecv_fn_t

typedef ncclResult_t(* ncclRecv_fn_t) (void *recvbuff, unsigned long count, ncclDataType_t datatype, int peer, ncclComm_t comm, hipStream_t stream)

Definition at line 141 of file api_trace.h.

◆ ncclRedOpCreatePreMulSum_fn_t

typedef ncclResult_t(* ncclRedOpCreatePreMulSum_fn_t) (ncclRedOp_t *op, void *scalar, ncclDataType_t datatype, ncclScalarResidence_t residence, ncclComm_t comm)

Definition at line 147 of file api_trace.h.

◆ ncclRedOpDestroy_fn_t

typedef ncclResult_t(* ncclRedOpDestroy_fn_t) (ncclRedOp_t op, ncclComm_t comm)

Definition at line 152 of file api_trace.h.

◆ ncclReduce_fn_t

typedef ncclResult_t(* ncclReduce_fn_t) (const void *sendbuff, void *recvbuff, unsigned long count, ncclDataType_t datatype, ncclRedOp_t op, int root, ncclComm_t comm, hipStream_t stream)

Definition at line 113 of file api_trace.h.

◆ ncclReduceScatter_fn_t

typedef ncclResult_t(* ncclReduceScatter_fn_t) (const void *sendbuff, void *recvbuff, unsigned long recvcount, ncclDataType_t datatype, ncclRedOp_t op, struct ncclComm *comm, hipStream_t stream)

Definition at line 121 of file api_trace.h.

◆ ncclScatter_fn_t

typedef ncclResult_t(* ncclScatter_fn_t) (const void *sendbuff, void *recvbuff, unsigned long recvcount, ncclDataType_t datatype, int root, ncclComm_t comm, hipStream_t stream)

Definition at line 128 of file api_trace.h.

◆ ncclSend_fn_t

typedef ncclResult_t(* ncclSend_fn_t) (const void *sendbuff, unsigned long count, ncclDataType_t datatype, int peer, ncclComm_t comm, hipStream_t stream)

Definition at line 135 of file api_trace.h.

◆ rccl_range_id_t

typedef uint64_t rccl_range_id_t

Definition at line 70 of file api_trace.h.