Merge pull request #1471 from lilohuang/main

alliepiper · web-flow · commit ed9945eb7a9d · 2021-07-21T12:33:07.000-04:00
Reduce scan_by_key memory consumption
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;
 

Original file line number	Diff line number	Diff line change
`@@ -108,7 +108,7 @@ __host__ __device__`
`108`	`108`	`AssociativeOperator binary_op)`
`109`	`109`	`{`
`110`	`110`	`using OutputType = typename thrust::iterator_traits<InputIterator2>::value_type;`
`111`		`- using HeadFlagType = thrust::detail::uint32_t;`
	`111`	`+ using HeadFlagType = thrust::detail::uint8_t;`
`112`	`112`
`113`	`113`	`const size_t n = last1 - first1;`
`114`	`114`
`@@ -204,7 +204,7 @@ __host__ __device__`
`204`	`204`	`AssociativeOperator binary_op)`
`205`	`205`	`{`
`206`	`206`	`using OutputType = T;`
`207`		`- using HeadFlagType = thrust::detail::uint32_t;`
	`207`	`+ using HeadFlagType = thrust::detail::uint8_t;`
`208`	`208`
`209`	`209`	`const size_t n = last1 - first1;`
`210`	`210`