CUTLASS 3.2 (#1024)

* CUTLASS 3.2
This commit is contained in:
ANIKET SHIVAM
2023-08-07 14:50:32 -10:00
committed by GitHub
parent a0d787b746
commit 4575443d44
392 changed files with 47559 additions and 7940 deletions

View File

@ -194,6 +194,146 @@ struct array
};
template <class T>
struct array<T, 0>
{
using value_type = T;
using size_type = size_t;
using difference_type = ptrdiff_t;
using reference = value_type&;
using const_reference = const value_type&;
using pointer = value_type*;
using const_pointer = const value_type*;
using const_iterator = const_pointer;
using iterator = const_iterator;
CUTE_HOST_DEVICE constexpr
reference operator[](size_type pos)
{
return begin()[pos];
}
CUTE_HOST_DEVICE constexpr
const_reference operator[](size_type pos) const
{
return begin()[pos];
}
CUTE_HOST_DEVICE constexpr
reference front()
{
return *begin();
}
CUTE_HOST_DEVICE constexpr
const_reference front() const
{
return *begin();
}
CUTE_HOST_DEVICE constexpr
reference back()
{
return *begin();
}
CUTE_HOST_DEVICE constexpr
const_reference back() const
{
return *begin();
}
CUTE_HOST_DEVICE constexpr
T* data()
{
return nullptr;
}
CUTE_HOST_DEVICE constexpr
T const* data() const
{
return nullptr;
}
CUTE_HOST_DEVICE constexpr
iterator begin()
{
return nullptr;
}
CUTE_HOST_DEVICE constexpr
const_iterator begin() const
{
return nullptr;
}
CUTE_HOST_DEVICE constexpr
const_iterator cbegin()
{
return nullptr;
}
CUTE_HOST_DEVICE constexpr
const_iterator cbegin() const
{
return nullptr;
}
CUTE_HOST_DEVICE constexpr
iterator end()
{
return nullptr;
}
CUTE_HOST_DEVICE constexpr
const_iterator end() const
{
return nullptr;
}
CUTE_HOST_DEVICE constexpr
const_iterator cend()
{
return nullptr;
}
CUTE_HOST_DEVICE constexpr
const_iterator cend() const
{
return nullptr;
}
CUTE_HOST_DEVICE constexpr
bool empty() const
{
return true;
}
CUTE_HOST_DEVICE constexpr
size_type size() const
{
return 0;
}
CUTE_HOST_DEVICE constexpr
size_type max_size() const
{
return 0;
}
CUTE_HOST_DEVICE constexpr
void fill(const T& value)
{}
CUTE_HOST_DEVICE constexpr
void clear()
{}
CUTE_HOST_DEVICE constexpr
void swap(array& other)
{}
};
template <class T, size_t N>
CUTE_HOST_DEVICE constexpr
bool operator==(array<T,N> const& lhs, array<T,N> const& rhs)
@ -227,6 +367,22 @@ void swap(array<T,N>& a, array<T,N>& b)
a.swap(b);
}
/// @return A cute::array of the elements of @c t in reverse order.
template <typename T, size_t N>
CUTE_HOST_DEVICE constexpr cute::array<T, N>
reverse(cute::array<T, N> const& t) {
if constexpr (N == 0u) {
return t;
}
else {
cute::array<T, N> t_r{};
for (size_t k = 0; k < N; ++k) {
t_r[k] = t[N - k - 1];
}
return t_r;
}
}
} // end cute
@ -274,7 +430,7 @@ namespace CUTE_STL_NAMESPACE
template <class T, size_t N>
struct tuple_size<cute::array<T,N>>
: cute::integral_constant<size_t, N>
: CUTE_STL_NAMESPACE::integral_constant<size_t, N>
{};
template <size_t I, class T, size_t N>
@ -285,7 +441,7 @@ struct tuple_element<I, cute::array<T,N>>
template <class T, size_t N>
struct tuple_size<const cute::array<T,N>>
: cute::integral_constant<size_t, N>
: CUTE_STL_NAMESPACE::integral_constant<size_t, N>
{};
template <size_t I, class T, size_t N>
@ -310,7 +466,7 @@ struct tuple_element;
template <class T, size_t N>
struct tuple_size<cute::array<T,N>>
: cute::integral_constant<size_t, N>
: CUTE_STL_NAMESPACE::integral_constant<size_t, N>
{};
template <size_t I, class T, size_t N>
@ -321,7 +477,7 @@ struct tuple_element<I, cute::array<T,N>>
template <class T, size_t N>
struct tuple_size<const cute::array<T,N>>
: cute::integral_constant<size_t, N>
: CUTE_STL_NAMESPACE::integral_constant<size_t, N>
{};
template <size_t I, class T, size_t N>
@ -330,5 +486,5 @@ struct tuple_element<I, const cute::array<T,N>>
using type = T;
};
} // end namepsace std
} // end namespace std
#endif // CUTE_STL_NAMESPACE_IS_CUDA_STD