code: def test_aot_compilation(donate):
      def f(x, y):
          return 2 * x + y
  
      x, y = jnp.array(3), 4
      lowered = eqx.filter_jit(f, donate=donate).lower(x, y)
      lowered.as_text()
      compiled = lowered.compile()
      compiled(x, y)
  
imports: import warnings
  from typing import Union
  import equinox
  import jax
  import jax.numpy
  import jax.random
  import jax.tree_util
  import pytest
  from .helpers import tree_allclose
label_desc: Use image node_id 15 for calling a global function with example usage: test_aot_compilation(donate) and returns: unknown