fix func bytecode

Author: makslevental

mlir/extras/dialects/ext/func.py

@@ -201,7 +201,9 @@ def __init__(
 
     def _is_decl(self):
         # magic constant found from looking at the code for an empty fn
-        if sys.version_info.minor == 13:
+        if sys.version_info.minor == 14:
+            return self.body_builder.__code__.co_code == b'\x80\x00R\x00#\x00'
+        elif sys.version_info.minor == 13:
             return self.body_builder.__code__.co_code == b"\x95\x00g\x00"
         elif sys.version_info.minor == 12:
             return self.body_builder.__code__.co_code == b"\x97\x00y\x00"

requirements-dev.txt

@@ -1,2 +1,3 @@
 black
 inflection
+pytest

tests/test_async.py

@@ -34,10 +34,6 @@ def test_simple_parfor(ctx: MLIRContext, backend: LLVMJITBackend):
         %initial = arith.constant 0 : index
         %step = arith.constant 1: index
         scf.parallel (%iv1, %iv2, %iv3, %iv4) = (%initial, %initial, %initial, %initial) to (%bound1, %bound2, %bound1, %bound2) step (%step, %step, %step, %step) {
-            vector.print %iv1 :index
-            vector.print %iv2 :index
-            vector.print %iv3 :index
-            vector.print %iv4 :index
             %e = arith.addi %iv3, %iv4 : index
             %d = arith.addi %e, %iv1 : index
             %b = arith.index_cast %d : index to i32

tests/test_func.py

@@ -46,7 +46,9 @@ def demo_fun1():
 def test_declare_byte_rep(ctx: MLIRContext):
     def demo_fun1(): ...
 
-    if sys.version_info.minor == 13:
+    if sys.version_info.minor == 14:
+        assert demo_fun1.__code__.co_code == b'\x80\x00R\x00#\x00'
+    elif sys.version_info.minor == 13:
         assert demo_fun1.__code__.co_code == b"\x95\x00g\x00"
     elif sys.version_info.minor == 12:
         assert demo_fun1.__code__.co_code == b"\x97\x00y\x00"