// Copyright 2023 The IREE Authors
//
// Licensed under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

IREE_NCCL_PFN_DECL(ncclGetVersion, int*)
IREE_NCCL_PFN_DECL(ncclGetUniqueId, ncclUniqueId*)
IREE_NCCL_PFN_DECL(ncclCommInitRankConfig, ncclComm_t*, int, ncclUniqueId, int,
                   ncclConfig_t*)
IREE_NCCL_PFN_DECL(ncclCommInitRank, ncclComm_t*, int, ncclUniqueId, int)
IREE_NCCL_PFN_DECL(ncclCommInitAll, ncclComm_t*, int, const int*)
IREE_NCCL_PFN_DECL(ncclCommSplit, ncclComm_t, int, int, ncclComm_t*,
                   ncclConfig_t*)
IREE_NCCL_PFN_DECL(ncclCommFinalize, ncclComm_t)
IREE_NCCL_PFN_DECL(ncclCommDestroy, ncclComm_t)
IREE_NCCL_PFN_DECL(ncclCommAbort, ncclComm_t)
IREE_NCCL_PFN_DECL_STR_RETURN(ncclGetErrorString, ncclResult_t)
IREE_NCCL_PFN_DECL_STR_RETURN(ncclGetLastError, ncclComm_t)
IREE_NCCL_PFN_DECL(ncclCommGetAsyncError, ncclComm_t, ncclResult_t*)
IREE_NCCL_PFN_DECL(ncclCommCount, const ncclComm_t, int*)
IREE_NCCL_PFN_DECL(ncclCommCuDevice, const ncclComm_t, int*)
IREE_NCCL_PFN_DECL(ncclCommUserRank, const ncclComm_t, int*)
IREE_NCCL_PFN_DECL(ncclRedOpCreatePreMulSum, ncclRedOp_t*, void*,
                   ncclDataType_t, ncclScalarResidence_t, ncclComm_t)
IREE_NCCL_PFN_DECL(ncclRedOpDestroy, ncclRedOp_t, ncclComm_t)
IREE_NCCL_PFN_DECL(ncclReduce, const void*, void*, size_t, ncclDataType_t,
                   ncclRedOp_t, int, ncclComm_t, cudaStream_t)
IREE_NCCL_PFN_DECL(ncclBcast, void*, size_t, ncclDataType_t, int, ncclComm_t,
                   cudaStream_t)
IREE_NCCL_PFN_DECL(ncclBroadcast, const void*, void*, size_t, ncclDataType_t,
                   int, ncclComm_t, cudaStream_t)
IREE_NCCL_PFN_DECL(ncclAllReduce, const void*, void*, size_t, ncclDataType_t,
                   ncclRedOp_t, ncclComm_t, cudaStream_t)
IREE_NCCL_PFN_DECL(ncclReduceScatter, const void*, void*, size_t,
                   ncclDataType_t, ncclRedOp_t, ncclComm_t, cudaStream_t)
IREE_NCCL_PFN_DECL(ncclAllGather, const void*, void*, size_t, ncclDataType_t,
                   ncclComm_t, cudaStream_t)
IREE_NCCL_PFN_DECL(ncclSend, const void*, size_t, ncclDataType_t, int,
                   ncclComm_t, cudaStream_t)
IREE_NCCL_PFN_DECL(ncclRecv, void*, size_t, ncclDataType_t, int, ncclComm_t,
                   cudaStream_t)
IREE_NCCL_PFN_DECL(ncclGroupStart)
IREE_NCCL_PFN_DECL(ncclGroupEnd)