Updates for CUTLASS 3.5.0 (#1468)
This commit is contained in:
@@ -41,7 +41,6 @@
|
||||
|
||||
#include <cute/pointer_base.hpp>
|
||||
#include <cute/pointer_swizzle.hpp>
|
||||
#include <cute/layout.hpp>
|
||||
namespace cute
|
||||
{
|
||||
|
||||
@@ -102,6 +101,8 @@ template <class P> // Found the gmem
|
||||
struct is_gmem<gmem_ptr<P>> : true_type {};
|
||||
template <class P> // Recurse on ::iterator, if possible
|
||||
struct is_gmem<P, void_t<typename P::iterator>> : is_gmem<typename P::iterator> {};
|
||||
template <class P>
|
||||
constexpr bool is_gmem_v = is_gmem<P>::value;
|
||||
|
||||
// Idempotent gmem tag on an iterator
|
||||
template <class Iterator>
|
||||
@@ -163,6 +164,8 @@ template <class P> // Found the smem
|
||||
struct is_smem<smem_ptr<P>> : true_type {};
|
||||
template <class P> // Recurse on ::iterator, if possible
|
||||
struct is_smem<P, void_t<typename P::iterator>> : is_smem<typename P::iterator> {};
|
||||
template <class P>
|
||||
constexpr bool is_smem_v = is_smem<P>::value;
|
||||
|
||||
// Idempotent smem tag on an iterator
|
||||
template <class Iterator>
|
||||
@@ -224,6 +227,8 @@ template <class T, class = void>
|
||||
struct is_rmem : bool_constant<not (is_gmem<T>::value || is_smem<T>::value)> {};
|
||||
template <class P>
|
||||
struct is_rmem<rmem_ptr<P>> : true_type {};
|
||||
template <class P>
|
||||
constexpr bool is_rmem_v = is_rmem<P>::value;
|
||||
|
||||
// Idempotent rmem tag on an iterator
|
||||
template <class Iterator>
|
||||
|
||||
Reference in New Issue
Block a user