33 righe
959 B
C
33 righe
959 B
C
|
/*************************************************************************
|
||
|
|
* Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved.
|
||
|
|
*
|
||
|
|
* See LICENSE.txt for license information
|
||
|
|
************************************************************************/
|
||
|
|
|
||
|
|
#ifndef NET_SOCKET_V1_H_
|
||
|
|
#define NET_SOCKET_V1_H_
|
||
|
|
|
||
|
|
#define NCCL_PROFILER_NET_SOCKET_VER 1
|
||
|
|
|
||
|
|
enum {
|
||
|
|
ncclProfileSocket = (1 << 0),
|
||
|
|
};
|
||
|
|
|
||
|
|
// The data structure version is encoded in the plugin identifier bitmask and
|
||
|
|
// passed to NCCL core through the profiler callback. NCCL copies the plugin
|
||
|
|
// identifier in the event descriptor before calling the profiler startEvent
|
||
|
|
// function. The profiler should inspect the plugin id to find out the source
|
||
|
|
// plugin as well as the version of the event struct
|
||
|
|
typedef struct {
|
||
|
|
uint8_t type; // event type (plugin defined)
|
||
|
|
union {
|
||
|
|
struct {
|
||
|
|
int fd;
|
||
|
|
int op;
|
||
|
|
size_t length;
|
||
|
|
} sock;
|
||
|
|
};
|
||
|
|
} ncclProfilerNetSockDescr_v1_t;
|
||
|
|
|
||
|
|
#endif
|