Update thrust/thrust/system/tbb/detail/reduce_by_key.inl

Co-authored-by: Georgii Evtushenko <[email protected]>
NVIDIA · Jul 25, 2024 · e781f99 · e781f99
1 parent b55eccd
commit e781f99
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/thrust/thrust/system/tbb/detail/reduce_by_key.inl b/thrust/thrust/system/tbb/detail/reduce_by_key.inl
@@ -64,9 +64,7 @@ inline L divide_ri(const L x, const R y)
 template <typename InputIterator, typename BinaryFunction, typename SFINAE = void>
 struct partial_sum_type
 {
-  using value_t = thrust::iterator_value_t<InputIterator>;
-  // TODO(bgruber): unify with CUB's accumulator_t
-  using type = ::cuda::std::__decay_t<typename ::cuda::std::__invoke_of<BinaryFunction, value_t, value_t>::type>;
+  using type = thrust::iterator_value_t<InputIterator>;
 };