55a4b22ad7
- Contains modifications to support AMD hardware - Adds unit tests
9 lines
319 B
C++
9 lines
319 B
C++
/*************************************************************************
|
|
* Copyright (c) 2019 Advanced Micro Devices, Inc. All rights reserved.
|
|
*
|
|
* See LICENSE.txt for license information
|
|
************************************************************************/
|
|
|
|
#define NCCL_OP 2
|
|
#include "device/all_reduce.cu"
|