1berry/ruby
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[PRISM] Improve dup_rest optimization targeting

Browse Source 
Part of implementing 3de20efc308cccc38bf9dacfffca6c626d039a06 in
prism_compile.c. Down to 2 failures from 30 failures in
test/ruby/test_allocation.rb.
This commit is contained in:
Alan Wu 2024-08-28 12:41:56 -04:00
parent ce6b0c4425
commit 7c9bcdf397
Notes: git 2024-08-28 17:34:57 +00:00 
Merged: https://github.com/ruby/ruby/pull/11479
1 changed files with 28 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
prism_compile.c
View File

@ -1818,10 +1818,24 @@ pm_setup_args(const pm_arguments_node_t *arguments_node, const pm_node_t *block,
VALUE dup_rest = Qtrue; VALUE dup_rest = Qtrue;
if (arguments_node != NULL) { if (arguments_node != NULL) {
if (arguments_node->arguments.size >= 2 && PM_NODE_TYPE_P(arguments_node->arguments.nodes[0], PM_SPLAT_NODE) && PM_NODE_TYPE_P(arguments_node->arguments.nodes[1], PM_KEYWORD_HASH_NODE)) { size_t arg_size = arguments_node->arguments.size;
const pm_node_list_t *args = &arguments_node->arguments;
// Calls like foo(1, *f, **hash) that use splat and kwsplat
// could be eligible for eliding duping the rest array (dup_reset=false).
if (arg_size >= 2
&& PM_NODE_TYPE_P(args->nodes[arg_size - 2], PM_SPLAT_NODE)
&& PM_NODE_TYPE_P(args->nodes[arg_size - 1], PM_KEYWORD_HASH_NODE)) {
// Reject calls that have more than one splat.
dup_rest = Qfalse; dup_rest = Qfalse;
for (size_t i = 0; i < arg_size - 2; i++) {
if (PM_NODE_TYPE_P(args->nodes[i], PM_SPLAT_NODE)) {
dup_rest = Qtrue;
break;
}
}
const pm_keyword_hash_node_t *keyword_arg = (const pm_keyword_hash_node_t *) arguments_node->arguments.nodes[1]; if (dup_rest == Qfalse) {
const pm_keyword_hash_node_t *keyword_arg = (const pm_keyword_hash_node_t *) args->nodes[arg_size - 1];
const pm_node_list_t *elements = &keyword_arg->elements; const pm_node_list_t *elements = &keyword_arg->elements;
if (PM_NODE_TYPE_P(elements->nodes[0], PM_ASSOC_NODE)) { if (PM_NODE_TYPE_P(elements->nodes[0], PM_ASSOC_NODE)) {
@ -1841,6 +1855,7 @@ pm_setup_args(const pm_arguments_node_t *arguments_node, const pm_node_t *block,
} }
} }
} }
}
VALUE initial_dup_rest = dup_rest; VALUE initial_dup_rest = dup_rest;