diff --git a/include/cutlass/array.h b/include/cutlass/array.h index 4f9c9185..61829091 100644 --- a/include/cutlass/array.h +++ b/include/cutlass/array.h @@ -454,6 +454,11 @@ public: return iterator(storage); } + CUTLASS_HOST_DEVICE + const_iterator begin() const { + return cbegin(); + } + CUTLASS_HOST_DEVICE const_iterator cbegin() const { return const_iterator(storage); @@ -464,6 +469,11 @@ public: return iterator(reinterpret_cast(storage + kStorageElements)); } + CUTLASS_HOST_DEVICE + const_iterator end() const { + return cend(); + } + CUTLASS_HOST_DEVICE const_iterator cend() const { return const_iterator(reinterpret_cast(storage + kStorageElements)); @@ -474,6 +484,11 @@ public: return reverse_iterator(reinterpret_cast(storage + kStorageElements)); } + CUTLASS_HOST_DEVICE + const_reverse_iterator rbegin() const { + return crbegin(); + } + CUTLASS_HOST_DEVICE const_reverse_iterator crbegin() const { return const_reverse_iterator(reinterpret_cast(storage + kStorageElements)); @@ -484,6 +499,11 @@ public: return reverse_iterator(reinterpret_cast(storage)); } + CUTLASS_HOST_DEVICE + const_reverse_iterator rend() const { + return crend(); + } + CUTLASS_HOST_DEVICE const_reverse_iterator crend() const { return const_reverse_iterator(reinterpret_cast(storage));