From bc75a0b58b91bc7a66a2457b1a8ec86e4dccb59f 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 58bebb72d7b..e057fe95abd 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,