summaryrefslogtreecommitdiff
path: root/yjit_iface.c
diff options
context:
space:
mode:
authorAlan Wu <XrXr@users.noreply.github.com>2021-11-26 18:00:42 -0500
committerGitHub <noreply@github.com>2021-11-26 18:00:42 -0500
commitb5b6ab4194f16e96ee5004288cc469ac1bca41a3 (patch)
treed4185b9e36c0863d9e82725234136ab082b84ed3 /yjit_iface.c
parent59439446d9432cabf106c700943b29773b4a264f (diff)
downloadruby-b5b6ab4194f16e96ee5004288cc469ac1bca41a3.tar.gz
YJIT: Add ability to exit to interpreter from stubs
Previously, YJIT assumed that it's always possible to generate a new basic block when servicing a stub in branch_stub_hit(). When YJIT is out of executable memory, for example, this assumption doesn't hold up. Add handling to branch_stub_hit() for servicing stubs without consuming more executable memory by adding a code path that exits to the interpreter at the location the branch stub represents. The new code path reconstructs interpreter state in branch_stub_hit() and then exits with a new snippet called `code_for_exit_from_stub` that returns `Qundef` from the YJIT native stack frame. As this change adds another place where we regenerate code from `branch_t`, extract the logic for it into a new function and call it regenerate_branch(). While we are at it, make the branch shrinking code path in branch_stub_hit() more explicit. This new functionality is hard to test without full support for out of memory conditions. To verify this change, I ran `RUBY_YJIT_ENABLE=1 make check -j12` with the following patch to stress test the new code path: ```diff diff --git a/yjit_core.c b/yjit_core.c index 4ab63d9806..5788b8c5ed 100644 --- a/yjit_core.c +++ b/yjit_core.c @@ -878,8 +878,12 @@ branch_stub_hit(branch_t *branch, const uint32_t target_idx, rb_execution_contex cb_set_write_ptr(cb, branch->end_addr); } +if (rand() < RAND_MAX/2) { // Compile the new block version p_block = gen_block_version(target, target_ctx, ec); +}else{ + p_block = NULL; +} if (!p_block && branch_modified) { // We couldn't generate a new block for the branch, but we modified the branch. ``` We can enable the new test along with other OOM tests once full support lands. Other small changes: * yjit_utils.c (print_str): Update to work with new native frame shape. Follow up for 8fa0ee4d404. * yjit_iface.c (rb_yjit_init): Run yjit_init_core() after yjit_init_codegen() so `cb` and `ocb` are available.
Diffstat (limited to 'yjit_iface.c')
-rw-r--r--yjit_iface.c5
1 files changed, 2 insertions, 3 deletions
diff --git a/yjit_iface.c b/yjit_iface.c
index a569128dce..917a32cbfa 100644
--- a/yjit_iface.c
+++ b/yjit_iface.c
@@ -1232,8 +1232,7 @@ rb_yjit_init(struct rb_yjit_options *options)
}
// If type propagation is disabled, max 1 version per block
- if (rb_yjit_opts.no_type_prop)
- {
+ if (rb_yjit_opts.no_type_prop) {
rb_yjit_opts.max_versions = 1;
}
@@ -1241,8 +1240,8 @@ rb_yjit_init(struct rb_yjit_options *options)
blocks_assuming_single_ractor_mode = st_init_numtable();
blocks_assuming_bops = st_init_numtable();
- yjit_init_core();
yjit_init_codegen();
+ yjit_init_core();
// YJIT Ruby module
mYjit = rb_define_module_under(rb_cRubyVM, "YJIT");