add jit_epilogue function