AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Unit test refactor (#500)

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

[ROCm/rccl commit: 29ad0f5fbe]
This commit is contained in:
gilbertlee-amd
2022-02-25 08:59:07 -07:00
committed by GitHub
parent dfa9b9e958
commit a182076a0e
98 changed files with 3684 additions and 5094 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/test/common/PrepDataFuncs.hpp
+26
View File
@@ -0,0 +1,26 @@
/*************************************************************************
* Copyright (c) 2022 Advanced Micro Devices, Inc. All rights reserved.
*
* See LICENSE.txt for license information
************************************************************************/
#pragma once
#include "ErrCode.hpp"
namespace RcclUnitTesting
{
class CollectiveArgs;
// Checks that enough memory has been allocated
ErrCode CheckAllocation(CollectiveArgs const& collArgs);
// Default PrepareData functions
// PrepareData functions are responsible for setting up input / expected for the given collArgs
ErrCode DefaultPrepareDataFunc(CollectiveArgs &collArgs);
ErrCode DefaultPrepData_Broadcast(CollectiveArgs &collArgs);
ErrCode DefaultPrepData_Reduce(CollectiveArgs &collArgs, bool const isAllReduce);
ErrCode DefaultPrepData_Gather(CollectiveArgs &collArgs, bool const isAllGather);
ErrCode DefaultPrepData_ReduceScatter(CollectiveArgs &collArgs);
ErrCode DefaultPrepData_Scatter(CollectiveArgs &collArgs);
ErrCode DefaultPrepData_AllToAll(CollectiveArgs &collArgs);
ErrCode DefaultPrepData_SendRecv(CollectiveArgs &collArgs);
}