2018-09-24 16:06:59 -07:00
|
|
|
/*************************************************************************
|
2020-05-12 14:40:18 -07:00
|
|
|
* Copyright (c) 2015-2020, NVIDIA CORPORATION. All rights reserved.
|
2018-09-24 16:06:59 -07:00
|
|
|
*
|
|
|
|
|
* See LICENSE.txt for license information
|
|
|
|
|
************************************************************************/
|
|
|
|
|
|
|
|
|
|
#ifndef NCCL_ENQUEUE_H_
|
|
|
|
|
#define NCCL_ENQUEUE_H_
|
|
|
|
|
|
2019-11-19 14:57:39 -08:00
|
|
|
#include "comm.h"
|
2018-09-24 16:06:59 -07:00
|
|
|
#include "group.h"
|
2019-11-19 14:57:39 -08:00
|
|
|
#include "collectives.h"
|
2019-03-14 19:39:20 -07:00
|
|
|
|
2018-12-13 15:56:12 -08:00
|
|
|
ncclResult_t ncclEnqueueCheck(struct ncclInfo* info);
|
2020-05-12 14:40:18 -07:00
|
|
|
ncclResult_t ncclCpuBarrierIn(struct ncclComm* comm, int* isLast);
|
|
|
|
|
ncclResult_t ncclCpuBarrierLast(struct ncclComm* comm);
|
|
|
|
|
ncclResult_t ncclCpuBarrierOut(struct ncclComm* comm);
|
|
|
|
|
ncclResult_t ncclBarrierEnqueue(struct ncclComm* comm);
|
|
|
|
|
ncclResult_t ncclBarrierEnqueueWait(struct ncclComm* comm);
|
|
|
|
|
ncclResult_t ncclEnqueueEvents(struct ncclComm* comm);
|
|
|
|
|
ncclResult_t ncclSaveKernel(struct ncclInfo* info);
|
2020-09-04 14:35:05 -07:00
|
|
|
ncclResult_t ncclSaveP2pKernel(struct ncclInfo* info);
|
|
|
|
|
ncclResult_t ncclSaveCommKernels(struct ncclComm* comm);
|
2018-09-24 16:06:59 -07:00
|
|
|
|
|
|
|
|
#endif // End include guard
|
