lpython.py: Support bitnot()

lcompilers · Jul 18, 2023 · f920fa6 · f920fa6
1 parent b6b09ec
commit f920fa6
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/runtime/lpython/lpython.py b/src/runtime/lpython/lpython.py
@@ -708,5 +708,10 @@ def __call__(self, *args, **kwargs):
             self.fn_name)
         return function(*args, **kwargs)
 
-def bitnot(x):
-    return ~x
+def bitnot(x, bitsize):
+    return (~x) % (2 ** bitsize)
+
+bitnot_u8 = lambda x: bitnot(x, 8)
+bitnot_u16 = lambda x: bitnot(x, 16)
+bitnot_u32 = lambda x: bitnot(x, 32)
+bitnot_u64 = lambda x: bitnot(x, 64)