fix allreduce tester (#385)

- use the reduce_psync buffers for synchronization in allreduce, not the
  barrier_psync.
- execute a wwg barrier after the allreduce operation. After internal
  discussion it was determined that it is required for correctness.
This commit is contained in:
Edgar Gabriel
2026-01-16 08:10:25 -06:00
committed by GitHub
parent 885e41ec62
commit 6f512e92a5
3 changed files with 7 additions and 30 deletions
@@ -83,7 +83,7 @@ __global__ void TeamReductionTest(int loop, int skip, long long int *start_time,
__shared__ rocshmem_ctx_t ctx;
int wg_id = get_flat_grid_id();
rocshmem_wg_ctx_create(ctx_type, &ctx);
rocshmem_wg_team_create_ctx(team, ctx_type, &ctx);
int n_pes = rocshmem_ctx_n_pes(ctx);