UTIL: Set rcache alignment to UCS_PGT_ADDR_ALIGN (openucx#996)

MamziB · Aug 20, 2024 · 7686688 · 7686688
1 parent 00d1220
commit 7686688
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/components/tl/mlx5/tl_mlx5_team.c b/src/components/tl/mlx5/tl_mlx5_team.c
@@ -313,7 +313,7 @@ ucc_status_t ucc_tl_mlx5_team_get_scores(ucc_base_team_t * tl_team,
  team_info.num_mem_types = 2;
  team_info.supported_mem_types = mt;
  team_info.supported_colls =
- (UCC_COLL_TYPE_ALLTOALL * (team->a2a_state == TL_MLX5_TEAM_STATE_ALLTOALL_READY) * 0) |
+ (UCC_COLL_TYPE_ALLTOALL * (team->a2a_state == TL_MLX5_TEAM_STATE_ALLTOALL_READY)) |
  UCC_COLL_TYPE_BCAST * (team->mcast_state == TL_MLX5_TEAM_STATE_MCAST_READY);
  team_info.size = UCC_TL_TEAM_SIZE(team);
 

diff --git a/src/utils/ucc_rcache.h b/src/utils/ucc_rcache.h
@@ -63,7 +63,7 @@ ucc_rcache_get(ucc_rcache_t *rcache, void *address, size_t length,
  ucs_status_t status;
 
 #ifdef UCS_HAVE_RCACHE_REGION_ALIGNMENT
- status = ucs_rcache_get(rcache, address, length, ucc_get_page_size(),
+ status = ucs_rcache_get(rcache, address, length, UCS_PGT_ADDR_ALIGN,
  PROT_READ | PROT_WRITE, arg, region_p);
 #else
  status = ucs_rcache_get(rcache, address, length,