summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--common.mk1
-rw-r--r--internal/cmdlineopt.h3
-rw-r--r--ruby.c33
-rw-r--r--version.c13
4 files changed, 29 insertions, 21 deletions
diff --git a/common.mk b/common.mk
index 934939ea4d..d6cbffe6d6 100644
--- a/common.mk
+++ b/common.mk
@@ -16771,6 +16771,7 @@ version.$(OBJEXT): $(hdrdir)/ruby.h
version.$(OBJEXT): $(hdrdir)/ruby/ruby.h
version.$(OBJEXT): $(hdrdir)/ruby/version.h
version.$(OBJEXT): $(top_srcdir)/internal/array.h
+version.$(OBJEXT): $(top_srcdir)/internal/cmdlineopt.h
version.$(OBJEXT): $(top_srcdir)/internal/compilers.h
version.$(OBJEXT): $(top_srcdir)/internal/gc.h
version.$(OBJEXT): $(top_srcdir)/internal/imemo.h
diff --git a/internal/cmdlineopt.h b/internal/cmdlineopt.h
index 71568a8745..bf52f1214b 100644
--- a/internal/cmdlineopt.h
+++ b/internal/cmdlineopt.h
@@ -36,6 +36,9 @@ typedef struct ruby_cmdline_options {
unsigned int do_split: 1;
unsigned int do_search: 1;
unsigned int setids: 2;
+#if USE_YJIT
+ unsigned int yjit: 1;
+#endif
} ruby_cmdline_options_t;
struct ruby_opt_message {
diff --git a/ruby.c b/ruby.c
index 640386fbda..fac0865302 100644
--- a/ruby.c
+++ b/ruby.c
@@ -73,7 +73,7 @@ STATIC_ASSERT(Qundef_1bit_from_Qnil, singlebit_only_p(Qundef^Qnil));
# define O_ACCMODE (O_RDONLY | O_WRONLY | O_RDWR)
#endif
-void Init_ruby_description(void);
+void Init_ruby_description(ruby_cmdline_options_t *opt);
#ifndef HAVE_STDLIB_H
char *getenv();
@@ -1614,13 +1614,8 @@ ruby_opt_init(ruby_cmdline_options_t *opt)
#if USE_MJIT
// rb_call_builtin_inits depends on RubyVM::MJIT.enabled?
- if (opt->mjit.on) {
+ if (opt->mjit.on)
mjit_enabled = true;
- // rb_threadptr_root_fiber_setup for the initial thread is called before rb_yjit_enabled_p()
- // or mjit_enabled becomes true, meaning jit_cont_new is skipped for the initial root fiber.
- // Therefore we need to call this again here to set the initial root fiber's jit_cont.
- rb_jit_cont_init(); // must be after mjit_enabled = true
- }
#endif
Init_ext(); /* load statically linked extensions before rubygems */
@@ -1628,11 +1623,20 @@ ruby_opt_init(ruby_cmdline_options_t *opt)
rb_call_builtin_inits();
ruby_init_prelude();
+ // Initialize JITs after prelude because JITing prelude is typically not optimal.
#if USE_MJIT
- // mjit_init is safe only after rb_call_builtin_inits defines RubyVM::MJIT::Compiler
+ // Also, mjit_init is safe only after rb_call_builtin_inits() defines RubyVM::MJIT::Compiler.
if (opt->mjit.on)
mjit_init(&opt->mjit);
#endif
+#if USE_YJIT
+ if (opt->yjit)
+ rb_yjit_init();
+#endif
+ // rb_threadptr_root_fiber_setup for the initial thread is called before rb_yjit_enabled_p()
+ // or mjit_enabled becomes true, meaning jit_cont_new is skipped for the initial root fiber.
+ // Therefore we need to call this again here to set the initial root fiber's jit_cont.
+ rb_jit_cont_init(); // must be after mjit_enabled = true and rb_yjit_init()
ruby_set_script_name(opt->script_name);
require_libraries(&opt->req_list);
@@ -1916,22 +1920,15 @@ process_options(int argc, char **argv, ruby_cmdline_options_t *opt)
#if USE_MJIT
if (FEATURE_SET_P(opt->features, mjit)) {
- opt->mjit.on = true; // set mjit.on for ruby_show_version() API and check to call mjit_init()
+ opt->mjit.on = true; // set opt->mjit.on for Init_ruby_description() and calling mjit_init()
}
#endif
#if USE_YJIT
if (FEATURE_SET_P(opt->features, yjit)) {
- rb_yjit_init();
- // rb_threadptr_root_fiber_setup for the initial thread is called before rb_yjit_enabled_p()
- // or mjit_enabled becomes true, meaning jit_cont_new is skipped for the initial root fiber.
- // Therefore we need to call this again here to set the initial root fiber's jit_cont.
- rb_jit_cont_init(); // must be after rb_yjit_init(), but before parsing options raises an exception.
+ opt->yjit = true; // set opt->yjit for Init_ruby_description() and calling rb_yjit_init()
}
#endif
-#if USE_MJIT
- mjit_opts.on = opt->mjit.on; /* used by Init_ruby_description(). mjit_init() still can't be called here. */
-#endif
- Init_ruby_description();
+ Init_ruby_description(opt);
if (opt->dump & (DUMP_BIT(version) | DUMP_BIT(version_v))) {
ruby_show_version();
if (opt->dump & DUMP_BIT(version)) return Qtrue;
diff --git a/version.c b/version.c
index 5ccc4cc508..a2d278f2db 100644
--- a/version.c
+++ b/version.c
@@ -9,6 +9,7 @@
**********************************************************************/
+#include "internal/cmdlineopt.h"
#include "ruby/ruby.h"
#include "version.h"
#include "vm_core.h"
@@ -121,13 +122,19 @@ Init_version(void)
}
#if USE_MJIT
-#define MJIT_OPTS_ON mjit_opts.on
+#define MJIT_OPTS_ON opt->mjit.on
#else
#define MJIT_OPTS_ON 0
#endif
+#if USE_YJIT
+#define YJIT_OPTS_ON opt->yjit
+#else
+#define YJIT_OPTS_ON 0
+#endif
+
void
-Init_ruby_description(void)
+Init_ruby_description(ruby_cmdline_options_t *opt)
{
VALUE description;
@@ -135,7 +142,7 @@ Init_ruby_description(void)
rb_dynamic_description = ruby_description_with_mjit;
description = MKSTR(description_with_mjit);
}
- else if (rb_yjit_enabled_p()) {
+ else if (YJIT_OPTS_ON) {
rb_dynamic_description = ruby_description_with_yjit;
description = MKSTR(description_with_yjit);
}