From c43c48af778730d67a2c2e342edf8074b5ded189 Mon Sep 17 00:00:00 2001 From: Marcel Koch Date: Tue, 17 Dec 2024 12:24:28 +0100 Subject: [PATCH] fixup! [dist] add round trip index mapping test --- test/distributed/index_map_kernels.cpp | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/test/distributed/index_map_kernels.cpp b/test/distributed/index_map_kernels.cpp index a697f35b899..68bff1b3db8 100644 --- 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(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(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(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,