[This follows on from: https://gcc.gnu.org/ml/gcc-patches/2019-09/msg00778.html https://gcc.gnu.org/ml/gcc-patches/2019-09/msg01456.html]
With the function ABI stuff, we can now support shrink-wrapping of non-leaf vector PCS functions. This is particularly useful if the vector PCS function calls an ordinary function on an error path, since we can then keep the extra saves and restores specific to that path too. Tested on aarch64-linux-gnu. OK to install? Richard 2019-09-25 Richard Sandiford <richard.sandif...@arm.com> gcc/ * config/aarch64/aarch64-protos.h (aarch64_use_simple_return_insn_p): Delete. * config/aarch64/aarch64.c (aarch64_components_for_bb): Check whether the block calls a function that clobbers more registers than the current function is allowed to. (aarch64_use_simple_return_insn_p): Delete. * config/aarch64/aarch64.md (simple_return): Remove condition. gcc/testsuite/ * gcc.target/aarch64/torture/simd-abi-9.c: New test. Index: gcc/config/aarch64/aarch64-protos.h =================================================================== --- gcc/config/aarch64/aarch64-protos.h 2019-09-21 13:56:09.083933336 +0100 +++ gcc/config/aarch64/aarch64-protos.h 2019-09-25 17:23:36.770504785 +0100 @@ -516,7 +516,6 @@ bool aarch64_split_dimode_const_store (r bool aarch64_symbolic_address_p (rtx); bool aarch64_uimm12_shift (HOST_WIDE_INT); bool aarch64_use_return_insn_p (void); -bool aarch64_use_simple_return_insn_p (void); const char *aarch64_mangle_builtin_type (const_tree); const char *aarch64_output_casesi (rtx *); Index: gcc/config/aarch64/aarch64.c =================================================================== --- gcc/config/aarch64/aarch64.c 2019-09-25 17:05:44.898231605 +0100 +++ gcc/config/aarch64/aarch64.c 2019-09-25 17:23:36.774504754 +0100 @@ -5976,13 +5976,30 @@ aarch64_components_for_bb (basic_block b sbitmap components = sbitmap_alloc (LAST_SAVED_REGNUM + 1); bitmap_clear (components); + /* Clobbered registers don't generate values in any meaningful sense, + since nothing after the clobber can rely on their value. And we can't + say that partially-clobbered registers are unconditionally killed, + because whether they're killed or not depends on the mode of the + value they're holding. Thus partially call-clobbered registers + appear in neither the kill set nor the gen set. + + Check manually for any calls that clobber more of a register than the + current function can. */ + function_abi_aggregator callee_abis; + rtx_insn *insn; + FOR_BB_INSNS (bb, insn) + if (CALL_P (insn)) + callee_abis.note_callee_abi (insn_callee_abi (insn)); + HARD_REG_SET extra_caller_saves = callee_abis.caller_save_regs (*crtl->abi); + /* GPRs are used in a bb if they are in the IN, GEN, or KILL sets. */ for (unsigned regno = 0; regno <= LAST_SAVED_REGNUM; regno++) if ((!call_used_or_fixed_reg_p (regno) || (simd_function && FP_SIMD_SAVED_REGNUM_P (regno))) - && (bitmap_bit_p (in, regno) - || bitmap_bit_p (gen, regno) - || bitmap_bit_p (kill, regno))) + && (TEST_HARD_REG_BIT (extra_caller_saves, regno) + || bitmap_bit_p (in, regno) + || bitmap_bit_p (gen, regno) + || bitmap_bit_p (kill, regno))) { unsigned regno2, offset, offset2; bitmap_set_bit (components, regno); @@ -6648,19 +6665,6 @@ aarch64_use_return_insn_p (void) return known_eq (cfun->machine->frame.frame_size, 0); } -/* Return false for non-leaf SIMD functions in order to avoid - shrink-wrapping them. Doing this will lose the necessary - save/restore of FP registers. */ - -bool -aarch64_use_simple_return_insn_p (void) -{ - if (aarch64_simd_decl_p (cfun->decl) && !crtl->is_leaf) - return false; - - return true; -} - /* Generate the epilogue instructions for returning from a function. This is almost exactly the reverse of the prolog sequence, except that we need to insert barriers to avoid scheduling loads that read Index: gcc/config/aarch64/aarch64.md =================================================================== --- gcc/config/aarch64/aarch64.md 2019-09-09 17:02:47.000000000 +0100 +++ gcc/config/aarch64/aarch64.md 2019-09-25 17:23:36.774504754 +0100 @@ -799,7 +799,7 @@ (define_expand "return" (define_insn "simple_return" [(simple_return)] - "aarch64_use_simple_return_insn_p ()" + "" "ret" [(set_attr "type" "branch")] ) Index: gcc/testsuite/gcc.target/aarch64/torture/simd-abi-9.c =================================================================== --- /dev/null 2019-09-17 11:41:18.176664108 +0100 +++ gcc/testsuite/gcc.target/aarch64/torture/simd-abi-9.c 2019-09-25 17:23:36.774504754 +0100 @@ -0,0 +1,48 @@ +/* { dg-do compile } */ +/* { dg-options "-fshrink-wrap -ffat-lto-objects" } */ +/* { dg-skip-if "" { *-*-* } { "-O0" } { "" } } */ +/* { dg-final { check-function-bodies "**" "" } } */ + +int callee (void); + +/* +** caller: +** ldr (w[0-9]+), \[x0\] +** cbn?z \1, [^\n]* +** ... +** ret +*/ +int __attribute__ ((aarch64_vector_pcs)) +caller (int *x) +{ + if (*x) + return callee () + 1; + else + return 0; +} + +/* { dg-final { scan-assembler {\sstp\tq8, q9} } } */ +/* { dg-final { scan-assembler {\sstp\tq10, q11} } } */ +/* { dg-final { scan-assembler {\sstp\tq12, q13} } } */ +/* { dg-final { scan-assembler {\sstp\tq14, q15} } } */ +/* { dg-final { scan-assembler {\sstp\tq16, q17} } } */ +/* { dg-final { scan-assembler {\sstp\tq18, q19} } } */ +/* { dg-final { scan-assembler {\sstp\tq20, q21} } } */ +/* { dg-final { scan-assembler {\sstp\tq22, q23} } } */ +/* { dg-final { scan-assembler {\sldp\tq8, q9} } } */ +/* { dg-final { scan-assembler {\sldp\tq10, q11} } } */ +/* { dg-final { scan-assembler {\sldp\tq12, q13} } } */ +/* { dg-final { scan-assembler {\sldp\tq14, q15} } } */ +/* { dg-final { scan-assembler {\sldp\tq16, q17} } } */ +/* { dg-final { scan-assembler {\sldp\tq18, q19} } } */ +/* { dg-final { scan-assembler {\sldp\tq20, q21} } } */ +/* { dg-final { scan-assembler {\sldp\tq22, q23} } } */ + +/* { dg-final { scan-assembler-not {\tstp\tq[0-7],} } } */ +/* { dg-final { scan-assembler-not {\tldp\tq[0-7],} } } */ +/* { dg-final { scan-assembler-not {\tstp\tq2[4-9],} } } */ +/* { dg-final { scan-assembler-not {\tldp\tq2[4-9],} } } */ +/* { dg-final { scan-assembler-not {\tstp\td} } } */ +/* { dg-final { scan-assembler-not {\tldp\td} } } */ +/* { dg-final { scan-assembler-not {\tstr\tq} } } */ +/* { dg-final { scan-assembler-not {\tldr\tq} } } */