Ameya Keshava Mallya
42d84317cf
git-subtree-dir: projects/rccl git-subtree-mainline:3fd8a0d393git-subtree-split:1f2f9f33ba
23 lines
729 B
C
23 lines
729 B
C
/*************************************************************************
|
|
* Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved.
|
|
*
|
|
* See LICENSE.txt for license information
|
|
************************************************************************/
|
|
|
|
#ifndef PROFILER_NET_H_
|
|
#define PROFILER_NET_H_
|
|
|
|
#define NCCL_PROFILER_NET_VER_BITS (16)
|
|
#define NCCL_PROFILER_NET_VER_MASK (~0U >> NCCL_PROFILER_NET_VER_BITS)
|
|
#define NCCL_PROFILER_NET_TYPE_MASK (~0U << NCCL_PROFILER_NET_VER_BITS)
|
|
|
|
typedef enum {
|
|
NCCL_PROFILER_NET_TYPE_IB = (1U << NCCL_PROFILER_NET_VER_BITS),
|
|
NCCL_PROFILER_NET_TYPE_SOCK = (2U << NCCL_PROFILER_NET_VER_BITS),
|
|
} ncclProfilerNetType;
|
|
|
|
#include "net_ib_v1.h"
|
|
#include "net_socket_v1.h"
|
|
|
|
#endif
|