acf0bc1c6e
[ROCm/rccl commit: 7a84c5dbb0]
18 lines
565 B
C++
18 lines
565 B
C++
/*************************************************************************
|
|
* Copyright (c) Advanced Micro Devices, Inc. All rights reserved.
|
|
*
|
|
* See LICENSE.txt for license information
|
|
************************************************************************/
|
|
|
|
#ifndef CALLCOLLECTIVEFORKED_H
|
|
#define CALLCOLLECTIVEFORKED_H
|
|
|
|
#include <vector>
|
|
|
|
namespace RcclUnitTesting
|
|
{
|
|
void callCollectiveForked(int nranks, int collID, const std::vector<int>& sendBuff, std::vector<int>& recvBuff, const std::vector<int>& expected, bool use_managed_mem = false);
|
|
}
|
|
|
|
#endif
|