procyberian/gcc
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

tree-optimization/114297 - SLP reduction with early break fix

Browse source 
The following makes sure to pass in the SLP node for the live stmts
we are generating the reduction epilogue for to
vect_create_epilog_for_reduction.  This follows the previous fix for
the non-SLP path.

	PR tree-optimization/114297
	* tree-vect-loop.cc (vectorizable_live_operation): Pass in the
	live stmts SLP node to vect_create_epilog_for_reduction.

	* gcc.dg/vect/vect-early-break_123-pr114297.c: New testcase.
This commit is contained in:
Richard Biener 2024-03-11 14:58:57 +01:00
parent c4e5789ced
commit c0c57246d5
2 changed files with 26 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
gcc/testsuite/gcc.dg/vect/vect-early-break_123-pr114297.c Normal file
View file

@ -0,0 +1,22 @@
/* { dg-do compile } */
/* { dg-add-options vect_early_break } */
/* { dg-require-effective-target vect_early_break } */
void h() __attribute__((__noreturn__));
struct Extremes {
int w;
int h;
};
struct Extremes *array;
int f(int num, int size1)
{
int sw = 0, sh = 0;
for (int i = 0; i < size1; ++i)
{
if (num - i == 0)
h();
sw += array[i].w;
sh += array[i].h;
}
return (sw) + (sh);
}

7
gcc/tree-vect-loop.cc
View file

@ -10729,17 +10729,18 @@ vectorizable_live_operation (vec_info *vinfo, stmt_vec_info stmt_info,
block, but we have to find an alternate exit first. */
if (LOOP_VINFO_EARLY_BREAKS (loop_vinfo))
{
slp_tree phis_node = slp_node ? slp_node_instance->reduc_phis : NULL;
for (auto exit : get_loop_exit_edges (LOOP_VINFO_LOOP (loop_vinfo)))
if (exit != LOOP_VINFO_IV_EXIT (loop_vinfo))
{
vect_create_epilog_for_reduction (loop_vinfo, reduc_info,
slp_node, slp_node_instance,
phis_node, slp_node_instance,
exit);
break;
}
if (LOOP_VINFO_EARLY_BREAKS_VECT_PEELED (loop_vinfo))
vect_create_epilog_for_reduction (loop_vinfo, reduc_info, slp_node,
slp_node_instance,
vect_create_epilog_for_reduction (loop_vinfo, reduc_info,
phis_node, slp_node_instance,
LOOP_VINFO_IV_EXIT (loop_vinfo));
}