UTIL: Set rcache alignment to UCS_PGT_ADDR_ALIGN (#996)

openucx · Aug 20, 2024 · 777df69 · 777df69
1 parent 985947a
commit 777df69
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,