quick fix in creating sub-communicator for hierarchy memory type

rapidsai · Oct 8, 2024 · 5aed03c · 5aed03c
1 parent 7680289
commit 5aed03c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cpp/src/wholememory/memory_handle.cpp b/cpp/src/wholememory/memory_handle.cpp
@@ -1778,7 +1778,7 @@ class hierarchy_wholememory_impl : public wholememory_impl {
 #endif
     } else {
       int world_rank = -1, local_size = -1;
-      wholememory_communicator_get_size(&world_rank, global_comm);
+      wholememory_communicator_get_rank(&world_rank, global_comm);
       wholememory_communicator_get_local_size(&local_size, global_comm);
       wholememory_split_communicator(
         &cross_comm_, global_comm, world_rank % local_size, world_rank / local_size);
@@ -2038,7 +2038,7 @@ wholememory_error_code_t create_wholememory(wholememory_handle_t* wholememory_ha
 #endif
       } else {
         int world_rank = -1, local_size = -1;
-        wholememory_communicator_get_size(&world_rank, comm);
+        wholememory_communicator_get_rank(&world_rank, comm);
         wholememory_communicator_get_local_size(&local_size, comm);
         wholememory_split_communicator(
           &local_comm, comm, world_rank / local_size, world_rank % local_size);