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

Update all_reduce algorithm to use internal put/get functions for updating pWrk and pSync arrays

Browse Source 
* Change log_stride calcualtions to stride calculations
* Update all_reduce example code to use team based interface
This commit is contained in:
avinashkethineedi
2024-10-28 22:10:18 +00:00
parent c22048112e
commit abec29bd6a
4 changed files with 35 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/functional_tests/tester_arguments.cpp
+5 -5
View File
@@ -136,11 +136,11 @@ void TesterArguments::get_rocshmem_arguments() {
myid = roc_shmem_my_pe();
TestType type = (TestType)algorithm;
if ((type != ReductionTestType) && (type != BarrierAllTestType) &&
(type != SyncAllTestType) && (type != SyncTestType) &&
(type != BroadcastTestType) && (type != AllToAllTestType) &&
(type != FCollectTestType) && (type != TeamReductionTestType) &&
(type != TeamBroadcastTestType) && (type != PingAllTestType)) {
if ((type != BarrierAllTestType) && (type != SyncAllTestType) &&
(type != SyncTestType) && (type != BroadcastTestType) &&
(type != AllToAllTestType) && (type != FCollectTestType) &&
(type != TeamReductionTestType) && (type != TeamBroadcastTestType) &&
(type != PingAllTestType)) {
if (numprocs != 2) {
if (myid == 0) {
std::cerr << "This test requires exactly two processes, we have "