feat: Check min_avm_version of intrinsic ops against target avm versi…

…on for compiler
algorandfoundation · Feb 1, 2024 · 2b3dea0 · 2b3dea0
1 parent be62082
commit 2b3dea0
Show file tree

Hide file tree

Showing 53 changed files with 210 additions and 48 deletions.
diff --git a/examples/amm/puya.log b/examples/amm/puya.log
diff --git a/examples/auction/puya.log b/examples/auction/puya.log
diff --git a/examples/calculator/puya.log b/examples/calculator/puya.log
diff --git a/examples/global_state/puya.log b/examples/global_state/puya.log
diff --git a/examples/hello_world/puya.log b/examples/hello_world/puya.log
diff --git a/examples/local_state/puya.log b/examples/local_state/puya.log
diff --git a/examples/voting/puya.log b/examples/voting/puya.log
diff --git a/scripts/generate_avm_ops.py b/scripts/generate_avm_ops.py
@@ -166,6 +166,7 @@ def generate_op_node(
         immediate_types=immediate_types,
         signature=signature,
         cost=cost,
+        min_avm_version=op.min_avm_version,
     )
     yield f"{op_name} = {data!r}"
     if op.doc:
@@ -212,6 +213,7 @@ class AVMOp(enum.StrEnum):
     immediate_types: Sequence[ImmediateKind]
     _signature: OpSignature | DynamicSignatures
     cost: int | None
+    min_avm_version: int
 
     def __new__(cls, data: AVMOpData | str) -> "AVMOp":
         # the weird union type on data && then assert,
@@ -225,6 +227,7 @@ def __new__(cls, data: AVMOpData | str) -> "AVMOp":
         obj.immediate_types = tuple(data.immediate_types)
         obj._signature = data.signature  # noqa: SLF001
         obj.cost = data.cost
+        obj.min_avm_version = data.min_avm_version
         return obj
 
     def get_signature(self, immediates: Sequence[str | int]) -> OpSignature:

diff --git a/src/puya/__main__.py b/src/puya/__main__.py
@@ -89,6 +89,7 @@ def main() -> None:
         type=LogLevel.from_string,
         choices=list(LogLevel),
     )
+    parser.add_argument("--target-avm-version", type=int, choices=[8, 9, 10])
 
     parser.add_argument("paths", type=Path, nargs="+", metavar="PATH")
     options = PuyaOptions()

diff --git a/src/puya/codegen/builder.py b/src/puya/codegen/builder.py
@@ -12,6 +12,7 @@
 )
 from puya.codegen.teal_writer import emit_memory_ir_as_teal
 from puya.context import CompileContext
+from puya.errors import CodeError
 from puya.ir import models
 from puya.ir.types_ import AVMBytesEncoding
 from puya.ir.visitor import IRVisitor
@@ -150,6 +151,13 @@ def visit_phi_argument(self, arg: models.PhiArgument) -> None:
         raise NotImplementedError
 
     def visit_intrinsic_op(self, intrinsic: models.Intrinsic) -> None:
+        if intrinsic.op.min_avm_version > self.context.options.target_avm_version:
+            raise CodeError(
+                f"Opcode {intrinsic.op.code} requires a min avm version of "
+                f"{intrinsic.op.min_avm_version} but the target avm version is"
+                f" {self.context.options.target_avm_version}",
+                intrinsic.source_location,
+            )
         for arg in intrinsic.args:
             arg.accept(self)
         self._add_op(

diff --git a/src/puya/codegen/teal_writer.py b/src/puya/codegen/teal_writer.py
@@ -84,7 +84,7 @@ def emit_memory_ir_as_teal(
             annotater.header(writer)
 
         writer.new_line()
-    writer.append_line("#pragma version 8")
+    writer.append_line(f"#pragma version {context.options.target_avm_version}")
     writer.new_line()
 
     for subroutine in subroutines: