fixup! [dist] add round trip index mapping test

MarcelKoch · MarcelKoch · commit c43c48af7787 · 2024-12-17T12:24:28.000+01:00
diff --git a/test/distributed/index_map_kernels.cpp b/test/distributed/index_map_kernels.cpp
@@ -561,9 +561,9 @@ TEST_F(IndexMap, RoundTripLocalWithLocalIndexSpace)
     auto global = gko::array<global_index_type>(exec);
 
     gko::kernels::GKO_DEVICE_NAMESPACE::index_map::map_to_global(
-        exec, to_device_const(part.get()), to_device_const(remote_global_idxs),
-        this_rank, query, gko::experimental::distributed::index_space::combined,
-        global);
+        exec, to_device_const(dpart.get()),
+        to_device_const(dremote_global_idxs), this_rank, query,
+        gko::experimental::distributed::index_space::combined, global);
     gko::kernels::GKO_DEVICE_NAMESPACE::index_map::map_to_local(
         exec, dpart.get(), dtarget_ids, to_device_const(dremote_global_idxs),
         this_rank, global,
@@ -600,9 +600,9 @@ TEST_F(IndexMap, RoundTripLocalWithNonLocalIndexSpace)
     auto global = gko::array<global_index_type>(exec);
 
     gko::kernels::GKO_DEVICE_NAMESPACE::index_map::map_to_global(
-        exec, to_device_const(part.get()), to_device_const(remote_global_idxs),
-        this_rank, query, gko::experimental::distributed::index_space::combined,
-        global);
+        exec, to_device_const(dpart.get()),
+        to_device_const(dremote_global_idxs), this_rank, query,
+        gko::experimental::distributed::index_space::combined, global);
     gko::kernels::GKO_DEVICE_NAMESPACE::index_map::map_to_local(
         exec, dpart.get(), dtarget_ids, to_device_const(dremote_global_idxs),
         this_rank, global,
@@ -649,9 +649,9 @@ TEST_F(IndexMap, RoundTripLocalWithCombinedIndexSpace)
     auto global = gko::array<global_index_type>(exec);
 
     gko::kernels::GKO_DEVICE_NAMESPACE::index_map::map_to_global(
-        exec, to_device_const(part.get()), to_device_const(remote_global_idxs),
-        this_rank, query, gko::experimental::distributed::index_space::combined,
-        global);
+        exec, to_device_const(dpart.get()),
+        to_device_const(dremote_global_idxs), this_rank, query,
+        gko::experimental::distributed::index_space::combined, global);
     gko::kernels::GKO_DEVICE_NAMESPACE::index_map::map_to_local(
         exec, dpart.get(), dtarget_ids, to_device_const(dremote_global_idxs),
         this_rank, global,
