Merge pull request #242 from visitorckw/fix-signed-integer-overflow

dboyliao · web-flow · commit 6aa081bf06c0 · 2024-09-06T23:54:03.000+08:00
Fix signed integer overflow by casting to uint32_t
diff --git a/src/uTensor/allocators/arenaAllocator.hpp b/src/uTensor/allocators/arenaAllocator.hpp
@@ -13,7 +13,7 @@
 namespace uTensor {
 
 //#define MSB_SET ~( ~( (T)0 ) >> 1 )
-#define MSB_SET (1 << (sizeof(uint32_t) * 8 - 1))
+#define MSB_SET ((uint32_t)1 << (sizeof(uint32_t) * 8 - 1))
 #define BLOCK_INACTIVE ~MSB_SET
 #define BLOCK_LENGTH_MASK ~MSB_SET
 #define BLOCK_ACTIVE MSB_SET
