2018-09-24 16:06:59 -07:00
|
|
|
/*************************************************************************
|
2022-01-07 06:39:55 -08:00
|
|
|
* Copyright (c) 2016-2022, NVIDIA CORPORATION. All rights reserved.
|
2018-09-24 16:06:59 -07:00
|
|
|
*
|
|
|
|
|
* See LICENSE.txt for license information
|
|
|
|
|
************************************************************************/
|
|
|
|
|
|
|
|
|
|
#ifndef NCCL_INT_NET_H_
|
|
|
|
|
#define NCCL_INT_NET_H_
|
|
|
|
|
|
|
|
|
|
#include "nccl.h"
|
|
|
|
|
#include "nccl_net.h"
|
2022-05-24 02:02:31 -07:00
|
|
|
#include "comm.h"
|
2022-01-07 06:39:55 -08:00
|
|
|
#include "checks.h"
|
2018-09-24 16:06:59 -07:00
|
|
|
|
|
|
|
|
typedef char ncclNetHandle_t[NCCL_NET_HANDLE_MAXSIZE];
|
|
|
|
|
|
2022-05-24 02:02:31 -07:00
|
|
|
ncclResult_t ncclNetInit(struct ncclComm* comm);
|
2024-06-11 01:28:01 -07:00
|
|
|
ncclResult_t ncclNetFinalize(struct ncclComm* comm);
|
2022-01-07 06:39:55 -08:00
|
|
|
|
2020-01-16 16:02:42 -08:00
|
|
|
// Test whether the current GPU support GPU Direct RDMA.
|
2022-05-24 02:02:31 -07:00
|
|
|
ncclResult_t ncclGpuGdrSupport(struct ncclComm* comm, int* gdrSupport);
|
2019-11-19 14:57:39 -08:00
|
|
|
|
2018-09-24 16:06:59 -07:00
|
|
|
extern ncclNet_t ncclNetIb;
|
|
|
|
|
extern ncclNet_t ncclNetSocket;
|
|
|
|
|
|
|
|
|
|
#endif
|
