From 67cff6d6cb9565f7f469d226afae580a8eb3451e Mon Sep 17 00:00:00 2001
From: Alex Light <allight@google.com>
Date: Thu, 9 Jan 2025 09:25:19 -0800
Subject: [PATCH] Update aot_main_wrapper template with new jit entrypoint
 arguments

The jit entrypoint signature changed but this was not updated. Update to match.

PiperOrigin-RevId: 713698012
---
 xls/jit/aot_main_cc.tmpl | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/xls/jit/aot_main_cc.tmpl b/xls/jit/aot_main_cc.tmpl
index 0bfd46ef9b..9094d450bd 100644
--- a/xls/jit/aot_main_cc.tmpl
+++ b/xls/jit/aot_main_cc.tmpl
@@ -12,7 +12,8 @@ void {{aot.extern_fn}}(const uint8_t* const* inputs,
                        uint8_t* const* outputs,
                        uint8_t* temp_buffer,
                        void* events,
-                       void* unused,
+                       void* unused,   // instance_context
+                       void* unused2,  // jit_runtime
                        int64_t continuation_point);
 {% if aot.write_result %}
 int write(int fd, void* v, size_t s);
@@ -54,7 +55,7 @@ int main(int argc, char** argv) {
   {{aot.extern_fn}}(
         arg_buffers, result_buffers, temp_buffer,
         /*interpreter_events=*/nullptr, /*callbacks=*/nullptr,
-        /*continuation_point=*/0);
+        /*runtime=*/nullptr, /*continuation_point=*/0);
   {% if aot.write_result %}
   write(1, result_buf, sizeof(result_buf));
   {% endif %}