src/collectives/device/all_reduce.cu


1
2
3
4
5
6
7
8
9
10
11

/*************************************************************************
 * Copyright (c) 2015-2019, NVIDIA CORPORATION. All rights reserved.
 *
 * See LICENSE.txt for license information
 ************************************************************************/

#include "all_reduce.h"
#include "common.h"
#include "collectives.h"

IMPL_COLL_R(ncclAllReduce, ncclCollAllReduce);