allow toggling between jit and bytecode

2 files changed, 9 insertions, 1 deletions

diff --git a/include/ejit/ejit.h b/include/ejit/ejit.h
index 3063ceb..c816a29 100644
--- a/include/ejit/ejit.h
+++ b/include/ejit/ejit.h
@@ -149,7 +149,10 @@ struct ejit_operand {
 struct ejit_func;
 
 struct ejit_func *ejit_create_func(enum ejit_type rtype, size_t argc, const struct ejit_operand args[argc]);
+
 void ejit_compile_func(struct ejit_func *f, size_t gpr, size_t fpr);
+void ejit_select_compile_func(struct ejit_func *f, size_t gpr, size_t fpr, bool try_jit);
+
 long ejit_run_func(struct ejit_func *f, size_t argc, struct ejit_arg args[argc]);
 double ejit_run_func_f(struct ejit_func *f, size_t argc, struct ejit_arg args[argc]);
 
diff --git a/src/ejit.c b/src/ejit.c
index 2e13e93..2040865 100644
--- a/src/ejit.c
+++ b/src/ejit.c
@@ -55,6 +55,11 @@ struct ejit_func *ejit_create_func(enum ejit_type rtype, size_t argc, const stru
 
 void ejit_compile_func(struct ejit_func *f, size_t gpr, size_t fpr)
 {
+	ejit_select_compile_func(f, gpr, fpr, true);
+}
+
+void ejit_select_compile_func(struct ejit_func *f, size_t gpr, size_t fpr, bool try_jit)
+{
 	/* emit a final end instruction in case user didn't do a return */
 	emit_insn_i(f, END, 0, 0, 0);
 
@@ -62,7 +67,7 @@ void ejit_compile_func(struct ejit_func *f, size_t gpr, size_t fpr)
 	f->fpr = fpr;
 
 	/* try to jit compile if possible */
-	if (ejit_compile(f))
+	if (try_jit && ejit_compile(f))
 		return;
 
 	/* otherwise, convert opcodes to address labels */