Merge pull request NVIDIA#1471 from lilohuang/main

Reduce scan_by_key memory consumption
MultithreadCorner · Jul 21, 2021 · ed9945e · ed9945e
2 parents ce483aa + a9143cf
commit ed9945e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/thrust/system/detail/generic/scan_by_key.inl b/thrust/system/detail/generic/scan_by_key.inl
@@ -108,7 +108,7 @@ __host__ __device__
                                        AssociativeOperator binary_op)
 {
   using OutputType = typename thrust::iterator_traits<InputIterator2>::value_type;
-  using HeadFlagType = thrust::detail::uint32_t;
+  using HeadFlagType = thrust::detail::uint8_t;
 
   const size_t n = last1 - first1;
 
@@ -204,7 +204,7 @@ __host__ __device__
                                        AssociativeOperator binary_op)
 {
   using OutputType = T;
-  using HeadFlagType = thrust::detail::uint32_t;
+  using HeadFlagType = thrust::detail::uint8_t;
 
   const size_t n = last1 - first1;