[fix] fix comparison operator for integer_subbyte (#1090)

SNU-ARC · Feb 7, 2024 · 8c600c6 · 8c600c6
1 parent 968dc6a
commit 8c600c6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -139,7 +139,7 @@ struct integer_subbyte {
         return !(rhs.storage < storage);
       }
     }
-    return storage < rhs.storage;
+    return storage <= rhs.storage;
   }
 
   /// Less than