Unit test refactor (#500)

Refactoring and consolidating single-process / multi-process unit testing

[ROCm/rccl commit: 29ad0f5fbe]
이 커밋은 다음에 포함됨:
gilbertlee-amd 작성하고 GitHub가 2022-02-25 08:59:07 -07:00를 커밋함
1 부모 dfa9b9e958
커밋 a182076a0e
98 파일이 변경됨
+3684 -5094

검색과 일치하는 파일 없음

+26
파일 보기
@@ -0,0 +1,26 @@
/*************************************************************************
* Copyright (c) 2022 Advanced Micro Devices, Inc. All rights reserved.
*
* See LICENSE.txt for license information
************************************************************************/
#include "TestBed.hpp"
namespace RcclUnitTesting
{
TEST(AllReduce, InPlace)
{
TestBed testBed;
// Configuration
std::vector<ncclFunc_t> const funcTypes = {ncclCollAllReduce};
std::vector<ncclDataType_t> const dataTypes = {ncclInt8, ncclInt32, ncclFloat32};
std::vector<ncclRedOp_t> const redOps = {ncclSum, ncclProd};
std::vector<int> const roots = {0};
std::vector<int> const numElements = {1048576, 53327, 1024};
std::vector<bool> const inPlaceList = {true};
std::vector<bool> const managedMemList = {false};
testBed.RunSimpleSweep(funcTypes, dataTypes, redOps, roots, numElements, inPlaceList, managedMemList);
testBed.Finalize();
}
}