2019-03-14 19:39:20 -07:00
|
|
|
/*************************************************************************
|
|
|
|
|
* Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved.
|
|
|
|
|
*
|
|
|
|
|
* See LICENSE.txt for license information
|
|
|
|
|
************************************************************************/
|
|
|
|
|
|
|
|
|
|
#ifndef NCCL_ARGCHECK_H_
|
|
|
|
|
#define NCCL_ARGCHECK_H_
|
|
|
|
|
|
|
|
|
|
#include "core.h"
|
2019-11-19 14:57:39 -08:00
|
|
|
#include "info.h"
|
2019-03-14 19:39:20 -07:00
|
|
|
|
|
|
|
|
ncclResult_t PtrCheck(void* ptr, const char* opname, const char* ptrname);
|
2024-03-26 06:08:55 -07:00
|
|
|
ncclResult_t CommCheck(struct ncclComm* ptr, const char* opname, const char* ptrname);
|
2019-03-14 19:39:20 -07:00
|
|
|
ncclResult_t ArgsCheck(struct ncclInfo* info);
|
2024-02-05 05:06:02 -08:00
|
|
|
ncclResult_t CudaPtrCheck(const void* pointer, struct ncclComm* comm, const char* ptrname, const char* opname);
|
2019-03-14 19:39:20 -07:00
|
|
|
|
|
|
|
|
#endif
|