27 라인
1.0 KiB
C
27 라인
1.0 KiB
C
|
/*************************************************************************
|
||
|
|
* Copyright (c) 2015-2019, NVIDIA CORPORATION. All rights reserved.
|
||
|
|
*
|
||
|
|
* See LICENSE.txt for license information
|
||
|
|
************************************************************************/
|
||
|
|
|
||
|
|
#include "tuner.h"
|
||
|
|
|
||
|
|
#define __hidden __attribute__ ((visibility("hidden")))
|
||
|
|
|
||
|
|
__hidden ncclResult_t pluginInit(size_t nRanks, size_t nNodes, ncclDebugLogger_t logFunction) { return ncclSuccess; }
|
||
|
|
|
||
|
|
__hidden ncclResult_t pluginGetCollInfo(ncclFunc_t collType, size_t nBytes,
|
||
|
|
int collNetSupport, int nvlsSupport, int numPipeOps,
|
||
|
|
int *algorithm, int *protocol, int* nChannels) { *algorithm = NCCL_ALGO_RING; *protocol = NCCL_PROTO_SIMPLE; return ncclSuccess; }
|
||
|
|
|
||
|
|
__hidden ncclResult_t pluginDestroy() { return ncclSuccess; }
|
||
|
|
|
||
|
|
#define PLUGIN_NAME "Example"
|
||
|
|
|
||
|
|
const ncclTuner_v1_t ncclTunerPlugin_v1 = {
|
||
|
|
.name = PLUGIN_NAME,
|
||
|
|
.init = pluginInit,
|
||
|
|
.getCollInfo = pluginGetCollInfo,
|
||
|
|
.destroy = pluginDestroy
|
||
|
|
};
|