2018-09-24 16:06:59 -07:00
|
|
|
/*************************************************************************
|
|
|
|
|
* Copyright (c) 2015-2018, NVIDIA CORPORATION. All rights reserved.
|
2019-08-12 18:23:23 +00:00
|
|
|
* Modifications Copyright (c) 2019 Advanced Micro Devices, Inc. All rights reserved.
|
2018-09-24 16:06:59 -07:00
|
|
|
*
|
|
|
|
|
* See LICENSE.txt for license information
|
|
|
|
|
************************************************************************/
|
|
|
|
|
|
|
|
|
|
#include "common.h"
|
|
|
|
|
#include "broadcast.h"
|
|
|
|
|
#include "collectives.h"
|
|
|
|
|
|
|
|
|
|
#define UNROLL 4
|
|
|
|
|
|
|
|
|
|
IMPL_COLL3(ncclBroadcast, copy, FuncSum, i8, int8_t, ncclCollBroadcast, ncclSum, ncclInt8);
|
