Final Fix

rapidsai · Sep 9, 2021 · a87aed9 · a87aed9
1 parent 810d234
commit a87aed9
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/cpp/src/groupby/hash/groupby.cu b/cpp/src/groupby/hash/groupby.cu
@@ -620,12 +620,8 @@ std::unique_ptr<table> groupby_null_templated(table_view const& keys,
                           stream,
                           mr);
 
-  return cudf::detail::gather(keys,
-                              gather_map.begin(),
-                              gather_map.begin() + gather_map.size(),
-                              out_of_bounds_policy::DONT_CHECK,
-                              stream,
-                              mr);
+  return cudf::detail::gather(
+    keys, gather_map.begin(), gather_map.end(), out_of_bounds_policy::DONT_CHECK, stream, mr);
 }
 
 }  // namespace