f9c3dc251e
Add local user buffer registration for NVLink SHARP. Add tuning plugin support. Increase net API to v7 to allow for device-side packet reordering; remove support for v4 plugins. Add support for RoCE ECE. Add support for C2C links. Better detect SHM allocation failures to avoid crash with Bus Error. Fix missing thread unlocks in bootstrap (Fixes #936). Disable network flush by default on H100. Move device code from src/collectives/device to src/device.
32 lines
1.1 KiB
C
32 lines
1.1 KiB
C
/*************************************************************************
|
|
* Copyright (c) 2023-2023, NVIDIA CORPORATION. All rights reserved.
|
|
*
|
|
* See LICENSE.txt for license information
|
|
************************************************************************/
|
|
|
|
#ifndef NET_DEVICE_H_
|
|
#define NET_DEVICE_H_
|
|
|
|
#include "net_device.h"
|
|
|
|
#define NCCL_NET_DEVICE_INVALID_VERSION 0x0
|
|
#define NCCL_NET_MTU_SIZE 4096
|
|
|
|
// Arbitrary version number - A given NCCL build will only be compatible with a single device networking plugin
|
|
// version. NCCL will check the supplied version number from net->getProperties() and compare to its internal version.
|
|
#define NCCL_NET_DEVICE_UNPACK_VERSION 0x7
|
|
|
|
typedef enum {NCCL_NET_DEVICE_HOST=0, NCCL_NET_DEVICE_UNPACK=1} ncclNetDeviceType;
|
|
|
|
typedef struct {
|
|
ncclNetDeviceType netDeviceType; // Network offload type
|
|
int netDeviceVersion; // Version number for network offload
|
|
void* handle;
|
|
size_t size;
|
|
int needsProxyProgress;
|
|
} ncclNetDeviceHandle_v7_t;
|
|
|
|
typedef ncclNetDeviceHandle_v7_t ncclNetDeviceHandle_t;
|
|
|
|
#endif
|