BRZ bigint 허용

aheui · Apr 5, 2024 · 2dc6764 · 2dc6764
1 parent adefff2
commit 2dc6764
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/aheui/aheui.py b/aheui/aheui.py
@@ -402,7 +402,8 @@ def mainloop(program, debug):
             elif op == c.OP_JMP:
                 jump = True
             elif op == c.OP_BRZ:
-                jump = 0 == selected.pop_longlong()
+                top = selected.pop()
+                jump = bigint.is_zero(top)
             else:
                 assert False
             if jump:

diff --git a/aheui/int/bigint.py b/aheui/int/bigint.py
@@ -54,3 +54,8 @@ def mod(r1, r2):
 @jit.elidable
 def ge(r1, r2):
     return r1.ge(r2)
+
+
+@jit.elidable
+def is_zero(r):
+    return r.sign == 0
diff --git a/aheui/int/smallint.py b/aheui/int/smallint.py
@@ -1,3 +1,9 @@
+try:
+    import builtins
+except ImportError:
+    builtins = __builtins__
+
+
 Int = int
 
 
@@ -43,3 +49,7 @@ def mod(r1, r2):
 
 def ge(r1, r2):
     return r1 >= r2
+
+
+def is_zero(r):
+    return r == 0