TL/MLX5: fix msgsize computation

openucx · Jul 13, 2023 · 33c1ee5 · 33c1ee5
1 parent 8d37b72
commit 33c1ee5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/components/tl/mlx5/alltoall/alltoall_coll.c b/src/components/tl/mlx5/alltoall/alltoall_coll.c
@@ -797,8 +797,9 @@ UCC_TL_MLX5_PROFILE_FUNC(ucc_status_t, ucc_tl_mlx5_alltoall_init,
         return UCC_ERR_NOT_SUPPORTED;
     }
 
-    msg_size = coll_args->args.src.info.count / UCC_TL_TEAM_SIZE(tl_team) *
-               ucc_dt_size(coll_args->args.src.info.datatype);
+    msg_size = (coll_args->args.src.info.count *
+                ucc_dt_size(coll_args->args.src.info.datatype)) /
+               UCC_TL_TEAM_SIZE(tl_team);
     if (!msg_size) {
         tl_trace(UCC_TL_TEAM_LIB(tl_team),
                  "msg size too short, reduces to nullop");