Just in time compilation with aarch64 target

.github/workflows/build-and-test-macos.yaml

@@ -47,6 +47,14 @@ jobs:
           otp: "28"
           cmake_opts_other: "-DAVM_DISABLE_JIT=OFF"
 
+        - os: "macos-14"
+          otp: "28"
+          cmake_opts_other: "-DAVM_DISABLE_JIT=OFF"
+
+        - os: "macos-15"
+          otp: "28"
+          cmake_opts_other: "-DAVM_DISABLE_JIT=OFF"
+
     steps:
     # Setup
     - name: "Checkout repo"

CMakeLists.txt

@@ -55,6 +55,10 @@ if (CMAKE_SYSTEM_PROCESSOR STREQUAL "x86_64")
     if (NOT AVM_DISABLE_JIT)
         set(AVM_JIT_TARGET_ARCH ${CMAKE_SYSTEM_PROCESSOR})
     endif()
+elseif(CMAKE_SYSTEM_PROCESSOR MATCHES "^arm64|aarch64$")
+    if (NOT AVM_DISABLE_JIT)
+        set(AVM_JIT_TARGET_ARCH "aarch64")
+    endif()
 else()
     if (NOT AVM_DISABLE_JIT)
         message("JIT is not supported on ${CMAKE_SYSTEM_PROCESSOR}")

libs/estdlib/src/code_server.erl

@@ -38,6 +38,7 @@
     code_chunk/1,
     atom_resolver/2,
     literal_resolver/2,
+    type_resolver/2,
     set_native_code/3
 ]).
 
@@ -126,6 +127,14 @@ atom_resolver(_Module, _Index) ->
 literal_resolver(_Module, _Index) ->
     erlang:nif_error(undefined).
 
+%% @doc Get a type from its index
+%% @return The type information
+%% @param Module module get a type from
+%% @param Index type index in the module
+-spec type_resolver(Module :: module(), Index :: non_neg_integer()) -> any().
+type_resolver(_Module, _Index) ->
+    erlang:nif_error(undefined).
+
 %% @doc Associate a native code stream with a module
 %% @return ok
 %% @param Module module to set the native code of
@@ -154,10 +163,16 @@ load(Module) ->
                         LiteralResolver = fun(Index) ->
                             code_server:literal_resolver(Module, Index)
                         end,
+                        TypeResolver = fun(Index) -> code_server:type_resolver(Module, Index) end,
                         Stream0 = jit:stream(jit_mmap_size(byte_size(Code))),
                         {BackendModule, BackendState0} = jit:backend(Stream0),
                         {LabelsCount, BackendState1} = jit:compile(
-                            Code, AtomResolver, LiteralResolver, BackendModule, BackendState0
+                            Code,
+                            AtomResolver,
+                            LiteralResolver,
+                            TypeResolver,
+                            BackendModule,
+                            BackendState0
                         ),
                         Stream1 = BackendModule:stream(BackendState1),
                         code_server:set_native_code(Module, LabelsCount, Stream1),

libs/jit/include/jit.hrl

@@ -21,6 +21,7 @@
 -define(JIT_FORMAT_VERSION, 1).
 
 -define(JIT_ARCH_X86_64, 1).
+-define(JIT_ARCH_AARCH64, 2).
 
 -define(JIT_VARIANT_PIC, 1).
 

libs/jit/src/CMakeLists.txt

@@ -27,6 +27,8 @@ set(ERLANG_MODULES
     jit_precompile
     jit_stream_binary
     jit_stream_mmap
+    jit_aarch64
+    jit_aarch64_asm
     jit_x86_64
     jit_x86_64_asm
 )