From 8c600c6e2c4e4cb1b9242a3a9a6b2c0d633f8e58 Mon Sep 17 00:00:00 2001
From: ZCHNO <zhengsizemax@outlook.com>
Date: Tue, 26 Sep 2023 14:26:12 -0700
Subject: [PATCH] [fix] fix comparison operator for integer_subbyte (#1090)

---
 include/cute/numeric/integer_subbyte.hpp | 2 +-
 include/cutlass/integer_subbyte.h        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/include/cute/numeric/integer_subbyte.hpp b/include/cute/numeric/integer_subbyte.hpp
index c20f485b..b10b45d8 100644
--- a/include/cute/numeric/integer_subbyte.hpp
+++ b/include/cute/numeric/integer_subbyte.hpp
@@ -111,7 +111,7 @@ struct integer_subbyte
     if (sign_mask_ & storage) {
       return !(rhs.storage < storage);
     } else {
-      return storage < rhs.storage;
+      return storage <= rhs.storage;
     }
   }
 
diff --git a/include/cutlass/integer_subbyte.h b/include/cutlass/integer_subbyte.h
index 0d5eb7bb..4c8ea36a 100644
--- a/include/cutlass/integer_subbyte.h
+++ b/include/cutlass/integer_subbyte.h
@@ -139,7 +139,7 @@ struct integer_subbyte {
         return !(rhs.storage < storage);
       }
     }
-    return storage < rhs.storage;
+    return storage <= rhs.storage;
   }
 
   /// Less than