From 9625e0c71cdceddd44ece33011ac5892830ed209 Mon Sep 17 00:00:00 2001 From: Seunghwa Kang Date: Tue, 20 Aug 2024 11:44:28 -0700 Subject: [PATCH] bug fix --- cpp/src/prims/detail/per_v_transform_reduce_e.cuh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cpp/src/prims/detail/per_v_transform_reduce_e.cuh b/cpp/src/prims/detail/per_v_transform_reduce_e.cuh index 44854a02164..a6d80be8557 100644 --- a/cpp/src/prims/detail/per_v_transform_reduce_e.cuh +++ b/cpp/src/prims/detail/per_v_transform_reduce_e.cuh @@ -1160,7 +1160,8 @@ void per_v_transform_reduce_e_edge_partition( auto segment_key_last = edge_partition_key_last; if constexpr (use_input_key) { segment_key_first += (*key_segment_offsets)[3]; - segment_key_last += (*key_segment_offsets)[4]; + segment_key_last = + segment_key_first + ((*key_segment_offsets)[4] - (*key_segment_offsets)[3]); } else { assert(segment_key_first == nullptr); assert(segment_key_last == nullptr);