Update iterator.h

NVIDIA · Dec 15, 2023 · ed0e53d · ed0e53d
1 parent 84eb23c
commit ed0e53d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/include/matx/core/iterator.h b/include/matx/core/iterator.h
@@ -286,7 +286,7 @@ struct RandomOperatorOutputIterator {
 template <typename OperatorType>
 struct BeginOffset {
   using self_type = BeginOffset<OperatorType>;
-  using value_type = typename OperatorType::scalar_type;
+  using value_type = index_t;
   // using stride_type = std::conditional_t<is_tensor_view_v<OperatorType>, typename OperatorType::desc_type::stride_type,
   //                         index_t>;
   using stride_type = index_t;
@@ -345,7 +345,7 @@ struct BeginOffset {
 template <typename OperatorType>
 struct EndOffset {
   using self_type = BeginOffset<OperatorType>;
-  using value_type = typename OperatorType::scalar_type;
+  using value_type = index_t;
   // using stride_type = std::conditional_t<is_tensor_view_v<OperatorType>, typename OperatorType::desc_type::stride_type,
   //                         index_t>;
   using stride_type = index_t;
@@ -411,4 +411,4 @@ auto  __MATX_INLINE__ __MATX_HOST__  end(Op &&op) {
   return RandomOperatorOutputIterator{static_cast<typename detail::base_type_t<Op>>(op), TotalSize(op)};
 }
 
-};
+};