Simplify GC step calls from on-trace code.

1 files changed, 3 insertions, 1 deletions

diff --git a/src/lj_gc.c b/src/lj_gc.c
index 18c07533..b97fb955 100644
--- a/src/lj_gc.c
+++ b/src/lj_gc.c
@@ -631,8 +631,10 @@ void LJ_FASTCALL lj_gc_step_fixtop(lua_State *L)
 
 #if LJ_HASJIT
 /* Perform multiple GC steps. Called from JIT-compiled code. */
-int LJ_FASTCALL lj_gc_step_jit(lua_State *L, MSize steps)
+int LJ_FASTCALL lj_gc_step_jit(global_State *g, MSize steps)
 {
+  lua_State *L = gco2th(gcref(g->jit_L));
+  L->base = mref(G(L)->jit_base, TValue);
   L->top = curr_topL(L);
   while (steps-- > 0 && lj_gc_step(L) == 0)
     ;