From 8890eba7dec45f0cad3b335c60bed7f94d4b6c3a Mon Sep 17 00:00:00 2001
From: Osayamen Aimuyo <oja7@cornell.edu>
Date: Sun, 14 Jul 2024 20:34:15 -0700
Subject: [PATCH] Add Integral Type Trait for integer_subbyte

---
 include/cute/numeric/int.hpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/include/cute/numeric/int.hpp b/include/cute/numeric/int.hpp
index 169e3a0e67..da738f4a2a 100644
--- a/include/cute/numeric/int.hpp
+++ b/include/cute/numeric/int.hpp
@@ -36,6 +36,8 @@
 #include <cstdint>
 #endif
 
+#include <cute/numeric/integral_constant.hpp>
+#include <cutlass/integer_subbyte.h>
 #include <cutlass/numeric_types.h>
 
 namespace cute
@@ -101,4 +103,7 @@ using uint_byte = uint_bit<8*N>;
 template <int N>
 using uint_byte_t = typename uint_byte<N>::type;
 
+template<int Bits, bool Signed>
+struct is_integral<cutlass::integer_subbyte<Bits, Signed>> : true_type {};
+
 } // namespace cute