From 777df69127a850312388626e7355d7b41e6a639e Mon Sep 17 00:00:00 2001 From: x41lakazam Date: Tue, 20 Aug 2024 09:33:27 +0300 Subject: [PATCH] UTIL: Set rcache alignment to UCS_PGT_ADDR_ALIGN (#996) --- src/components/tl/mlx5/tl_mlx5_team.c | 2 +- src/utils/ucc_rcache.h | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/components/tl/mlx5/tl_mlx5_team.c b/src/components/tl/mlx5/tl_mlx5_team.c index edc0b95d90..614ead348b 100644 --- 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 index 3e89396a93..1995e11ba8 100644 --- 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,