2022-11-21 06:03:27 -08:00
|
|
|
/*
|
|
|
|
|
* Copyright (c) 2017-2022, NVIDIA CORPORATION. All rights reserved.
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
#ifndef NCCL_NET_H_
|
|
|
|
|
#define NCCL_NET_H_
|
|
|
|
|
|
|
|
|
|
#include <stdint.h>
|
|
|
|
|
#include <stdlib.h>
|
|
|
|
|
|
2024-06-11 01:28:01 -07:00
|
|
|
#include "common.h"
|
2022-11-21 06:03:27 -08:00
|
|
|
#include "err.h"
|
|
|
|
|
|
|
|
|
|
#define NCCL_NET_HANDLE_MAXSIZE 128
|
|
|
|
|
|
|
|
|
|
#define NCCL_PTR_HOST 0x1
|
|
|
|
|
#define NCCL_PTR_CUDA 0x2
|
|
|
|
|
#define NCCL_PTR_DMABUF 0x4
|
|
|
|
|
|
|
|
|
|
// Maximum number of requests per comm object
|
2024-02-05 05:06:02 -08:00
|
|
|
#define NCCL_NET_MAX_REQUESTS 32
|
2022-11-21 06:03:27 -08:00
|
|
|
|
2024-02-05 05:06:02 -08:00
|
|
|
#include "net_v8.h"
|
2023-09-26 05:47:28 -07:00
|
|
|
#include "net_v7.h"
|
2022-11-21 06:03:27 -08:00
|
|
|
#include "net_v6.h"
|
|
|
|
|
#include "net_v5.h"
|
|
|
|
|
#include "net_v4.h"
|
|
|
|
|
#include "net_v3.h"
|
|
|
|
|
#include "net_v2.h"
|
|
|
|
|
|
|
|
|
|
#endif // end include guard
|
