UTIL: Set rcache alignment to UCS_PGT_ADDR_ALIGN

openucx · Jul 14, 2024 · e945fca · e945fca
1 parent 37adf47
commit e945fca
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/components/tl/mlx5/tl_mlx5_team.c b/src/components/tl/mlx5/tl_mlx5_team.c
@@ -242,7 +242,7 @@ ucc_status_t ucc_tl_mlx5_team_create_test(ucc_base_team_t *team)
     }
 
     ucc_assert(tl_team->global_sync_req == NULL);
-
+     
     if (tl_team->mcast_state == TL_MLX5_TEAM_STATE_MCAST_CTX_CHECK &&
         tl_team->a2a_state == TL_MLX5_TEAM_STATE_ALLTOALL_CTX_CHECK) {
         // check if ctx is ready for a2a and mcast
@@ -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,