Allow future function label assignment.

src/starkware/cairo/lang/compiler/ast/expr.py

@@ -406,7 +406,7 @@ def to_expr_str(self):
         return self.identifier.to_expr_str()
 
     @property
-    def locaion(self):
+    def location(self):
         return self.identifier.location
 
     def get_children(self) -> Sequence[Optional[AstNode]]:

src/starkware/cairo/lang/compiler/preprocessor/compound_expressions.py

@@ -12,6 +12,7 @@
     ExprDeref,
     Expression,
     ExprHint,
+    ExprFutureLabel,
     ExprIdentifier,
     ExprNeg,
     ExprOperator,
@@ -161,6 +162,12 @@ def rewrite_ExprDeref(self, expr: ExprDeref, sim: SimplicityLevel):
         )
         return expr if sim is SimplicityLevel.OPERATION else self.wrap(expr)
 
+    def rewrite_ExprFutureLabel(self, expr: ExprFutureLabel, sim: SimplicityLevel):
+        # Treat this as a constant.
+        if sim in [SimplicityLevel.DEREF_CONST, SimplicityLevel.OPERATION]:
+            return expr
+        return self.wrap(expr)
+
     def rewrite_ExprHint(self, expr: ExprHint, sim: SimplicityLevel):
         return self.wrap(expr)
 

src/starkware/cairo/lang/compiler/preprocessor/preprocessor.py

@@ -1899,7 +1899,7 @@ def get_variable(self, var: ExprIdentifier):
         assert identifier_definition is not None
 
         if isinstance(identifier_definition, FutureIdentifierDefinition):
-            if identifier_definition.identifier_type == LabelDefinition:
+            if identifier_definition.identifier_type in [LabelDefinition, FunctionDefinition]:
                 # Allow future label assignment.
                 return ExprFutureLabel(identifier=var)
             raise PreprocessorError(

src/starkware/cairo/lang/compiler/preprocessor/preprocessor_test.py

@@ -5,6 +5,7 @@
 from starkware.cairo.lang.compiler.error_handling import LocationError
 from starkware.cairo.lang.compiler.identifier_definition import (
     ConstDefinition,
+    FunctionDefinition,
     LabelDefinition,
     ReferenceDefinition,
 )
@@ -167,6 +168,37 @@ def test_assign_future_label():
     )
 
 
+def test_assign_future_function_label():
+    code = """\
+g(f)
+g((f + 1) * 2 + 3)
+
+func f() -> ():
+    ret
+end
+func g(x: felt) -> ():
+    ret
+end
+"""
+    program = preprocess_str(code=code, prime=PRIME)
+    f_definition = program.identifiers.get_by_full_name(ScopedName.from_string("test_scope.f"))
+    assert isinstance(f_definition, FunctionDefinition)
+    assert (
+        program.format()
+        == f"""\
+[ap] = {f_definition.pc}; ap++
+call rel 13
+[ap] = {f_definition.pc}; ap++
+[ap] = [ap + (-1)] + 1; ap++
+[ap] = [ap + (-1)] * 2; ap++
+[ap] = [ap + (-1)] + 3; ap++
+call rel 3
+ret
+ret
+"""
+    )
+
+
 def test_temporary_variable():
     code = """\
 struct T: