diff mbox

[AArch64] Separate shrink wrapping hooks implementation

Message ID 5829C958.2010601@foss.arm.com
State Superseded
Headers show

Commit Message

Kyrill Tkachov Nov. 14, 2016, 2:25 p.m. UTC
On 11/11/16 15:31, Kyrill Tkachov wrote:
>

> On 11/11/16 10:17, Kyrill Tkachov wrote:

>>

>> On 10/11/16 23:39, Segher Boessenkool wrote:

>>> On Thu, Nov 10, 2016 at 02:42:24PM -0800, Andrew Pinski wrote:

>>>> On Thu, Nov 10, 2016 at 6:25 AM, Kyrill Tkachov

>>>>> I ran SPEC2006 on a Cortex-A72. Overall scores were neutral but there were

>>>>> some interesting swings.

>>>>> 458.sjeng     +1.45%

>>>>> 471.omnetpp   +2.19%

>>>>> 445.gobmk     -2.01%

>>>>>

>>>>> On SPECFP:

>>>>> 453.povray    +7.00%

>>>>

>>>> Wow, this looks really good.  Thank you for implementing this.  If I

>>>> get some time I am going to try it out on other processors than A72

>>>> but I doubt I have time any time soon.

>>> I'd love to hear what causes the slowdown for gobmk as well, btw.

>>

>> I haven't yet gotten a direct answer for that (through performance analysis tools)

>> but I have noticed that load/store pairs are not generated as aggressively as I hoped.

>> They are being merged by the sched fusion pass and peepholes (which runs after this)

>> but it still misses cases. I've hacked the SWS hooks to generate pairs explicitly and that

>> increases the number of pairs and helps code size to boot. It complicates the logic of

>> the hooks a bit but not too much.

>>

>> I'll make those changes and re-benchmark, hopefully that

>> will help performance.

>>

>

> And here's a version that explicitly emits pairs. I've looked at assembly codegen on SPEC2006

> and it generates quite a few more LDP/STP pairs than the original version.

> I kicked off benchmarks over the weekend to see the effect.

> Andrew, if you want to try it out (more benchmarking and testing always welcome) this is the

> one to try.

>


And I discovered over the weekend that gamess and wrf have validation errors.
This version runs correctly.
SPECINT results were fine though and there is even a small overall gain due to
sjeng and omnetpp. However, gobmk still has the regression.
I'll rerun SPECFP with this patch (it's really just a small bugfix over the previous version)
and get on with analysing gobmk.

Thanks,
Kyrill

2016-11-11  Kyrylo Tkachov  <kyrylo.tkachov@arm.com>

     * config/aarch64/aarch64.h (machine_function): Add
     reg_is_wrapped_separately field.
     * config/aarch64/aarch64.c (emit_set_insn): Change return type to
     rtx_insn *.
     (aarch64_save_callee_saves): Don't save registers that are wrapped
     separately.
     (aarch64_restore_callee_saves): Don't restore registers that are
     wrapped separately.
     (offset_9bit_signed_unscaled_p, offset_12bit_unsigned_scaled_p,
     aarch64_offset_7bit_signed_scaled_p): Move earlier in the file.
     (aarch64_get_separate_components): New function.
     (aarch64_get_next_set_bit): Likewise.
     (aarch64_components_for_bb): Likewise.
     (aarch64_disqualify_components): Likewise.
     (aarch64_emit_prologue_components): Likewise.
     (aarch64_emit_epilogue_components): Likewise.
     (aarch64_set_handled_components): Likewise.
     (TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS,
     TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB,
     TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS,
     TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS,
     TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS,
     TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS): Define.

Comments

Kyrill Tkachov Nov. 17, 2016, 2:22 p.m. UTC | #1
On 14/11/16 14:25, Kyrill Tkachov wrote:
>

> On 11/11/16 15:31, Kyrill Tkachov wrote:

>>

>> On 11/11/16 10:17, Kyrill Tkachov wrote:

>>>

>>> On 10/11/16 23:39, Segher Boessenkool wrote:

>>>> On Thu, Nov 10, 2016 at 02:42:24PM -0800, Andrew Pinski wrote:

>>>>> On Thu, Nov 10, 2016 at 6:25 AM, Kyrill Tkachov

>>>>>> I ran SPEC2006 on a Cortex-A72. Overall scores were neutral but there were

>>>>>> some interesting swings.

>>>>>> 458.sjeng     +1.45%

>>>>>> 471.omnetpp   +2.19%

>>>>>> 445.gobmk     -2.01%

>>>>>>

>>>>>> On SPECFP:

>>>>>> 453.povray    +7.00%

>>>>>

>>>>> Wow, this looks really good.  Thank you for implementing this.  If I

>>>>> get some time I am going to try it out on other processors than A72

>>>>> but I doubt I have time any time soon.

>>>> I'd love to hear what causes the slowdown for gobmk as well, btw.

>>>

>>> I haven't yet gotten a direct answer for that (through performance analysis tools)

>>> but I have noticed that load/store pairs are not generated as aggressively as I hoped.

>>> They are being merged by the sched fusion pass and peepholes (which runs after this)

>>> but it still misses cases. I've hacked the SWS hooks to generate pairs explicitly and that

>>> increases the number of pairs and helps code size to boot. It complicates the logic of

>>> the hooks a bit but not too much.

>>>

>>> I'll make those changes and re-benchmark, hopefully that

>>> will help performance.

>>>

>>

>> And here's a version that explicitly emits pairs. I've looked at assembly codegen on SPEC2006

>> and it generates quite a few more LDP/STP pairs than the original version.

>> I kicked off benchmarks over the weekend to see the effect.

>> Andrew, if you want to try it out (more benchmarking and testing always welcome) this is the

>> one to try.

>>

>

> And I discovered over the weekend that gamess and wrf have validation errors.

> This version runs correctly.

> SPECINT results were fine though and there is even a small overall gain due to

> sjeng and omnetpp. However, gobmk still has the regression.

> I'll rerun SPECFP with this patch (it's really just a small bugfix over the previous version)

> and get on with analysing gobmk.

>


After looking at the gobmk performance with performance counters it looks like more icache pressure.
I see an increase in misses.
This looks to me like an effect of code size increase, though it is not that large an increase (0.4% with SWS).
Branch mispredicts also go up a bit but not as much as icache misses.
I don't think there's anything we can do here, or at least that this patch can do about it.
Overall, there's a slight improvement in SPECINT, even with the gobmk regression and a slightly larger improvement
on SPECFP due to povray.

Segher, one curious artifact I spotted while looking at codegen differences in gobmk was a case where we fail
to emit load-pairs as effectively in the epilogue and its preceeding basic block.
So before we had this epilogue:
.L43:
     ldp    x21, x22, [sp, 16]
     ldp    x23, x24, [sp, 32]
     ldp    x25, x26, [sp, 48]
     ldp    x27, x28, [sp, 64]
     ldr    x30, [sp, 80]
     ldp    x19, x20, [sp], 112
     ret

and I see this becoming (among numerous other changes in the function):

.L69:
     ldp    x21, x22, [sp, 16]
     ldr    x24, [sp, 40]
.L43:
     ldp    x25, x26, [sp, 48]
     ldp    x27, x28, [sp, 64]
     ldr    x23, [sp, 32]
     ldr    x30, [sp, 80]
     ldp    x19, x20, [sp], 112
     ret

So this is better in the cases where we jump straight into .L43 because we load fewer registers
but worse when we jump to or fallthrough to .L69 because x23 and x24 are now restored using two loads
rather than a single load-pair. This hunk isn't critical to performance in gobmk though.

Given that there is an overall gain is this ok for trunk?
https://gcc.gnu.org/ml/gcc-patches/2016-11/msg01352.html

Thanks,
Kyrill

>

> 2016-11-11  Kyrylo Tkachov  <kyrylo.tkachov@arm.com>

>

>     * config/aarch64/aarch64.h (machine_function): Add

>     reg_is_wrapped_separately field.

>     * config/aarch64/aarch64.c (emit_set_insn): Change return type to

>     rtx_insn *.

>     (aarch64_save_callee_saves): Don't save registers that are wrapped

>     separately.

>     (aarch64_restore_callee_saves): Don't restore registers that are

>     wrapped separately.

>     (offset_9bit_signed_unscaled_p, offset_12bit_unsigned_scaled_p,

>     aarch64_offset_7bit_signed_scaled_p): Move earlier in the file.

>     (aarch64_get_separate_components): New function.

>     (aarch64_get_next_set_bit): Likewise.

>     (aarch64_components_for_bb): Likewise.

>     (aarch64_disqualify_components): Likewise.

>     (aarch64_emit_prologue_components): Likewise.

>     (aarch64_emit_epilogue_components): Likewise.

>     (aarch64_set_handled_components): Likewise.

>     (TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS,

>     TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB,

>     TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS,

>     TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS,

>     TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS,

>     TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS): Define.

>
Segher Boessenkool Nov. 17, 2016, 2:44 p.m. UTC | #2
Hi Kyrill,

On Thu, Nov 17, 2016 at 02:22:08PM +0000, Kyrill Tkachov wrote:
> >>>>>>I ran SPEC2006 on a Cortex-A72. Overall scores were neutral but there 

> >>>>>>were

> >>>>>>some interesting swings.

> >>>>>>458.sjeng     +1.45%

> >>>>>>471.omnetpp   +2.19%

> >>>>>>445.gobmk     -2.01%

> >>>>>>

> >>>>>>On SPECFP:

> >>>>>>453.povray    +7.00%


> After looking at the gobmk performance with performance counters it looks 

> like more icache pressure.

> I see an increase in misses.

> This looks to me like an effect of code size increase, though it is not 

> that large an increase (0.4% with SWS).


Right.  I don't see how to improve on this (but see below); ideas welcome :-)

> Branch mispredicts also go up a bit but not as much as icache misses.


I don't see that happening -- for some testcases we get unlucky and have
more branch predictor aliasing, and for some we have less, it's pretty
random.  Some testcases are really sensitive to this.

> I don't think there's anything we can do here, or at least that this patch 

> can do about it.

> Overall, there's a slight improvement in SPECINT, even with the gobmk 

> regression and a slightly larger improvement

> on SPECFP due to povray.


And that is for only the "normal" GPRs, not LR or FP yet, right?

> Segher, one curious artifact I spotted while looking at codegen differences 

> in gobmk was a case where we fail

> to emit load-pairs as effectively in the epilogue and its preceeding basic 

> block.

> So before we had this epilogue:

> .L43:

>     ldp    x21, x22, [sp, 16]

>     ldp    x23, x24, [sp, 32]

>     ldp    x25, x26, [sp, 48]

>     ldp    x27, x28, [sp, 64]

>     ldr    x30, [sp, 80]

>     ldp    x19, x20, [sp], 112

>     ret

> 

> and I see this becoming (among numerous other changes in the function):

> 

> .L69:

>     ldp    x21, x22, [sp, 16]

>     ldr    x24, [sp, 40]

> .L43:

>     ldp    x25, x26, [sp, 48]

>     ldp    x27, x28, [sp, 64]

>     ldr    x23, [sp, 32]

>     ldr    x30, [sp, 80]

>     ldp    x19, x20, [sp], 112

>     ret

> 

> So this is better in the cases where we jump straight into .L43 because we 

> load fewer registers

> but worse when we jump to or fallthrough to .L69 because x23 and x24 are 

> now restored using two loads

> rather than a single load-pair. This hunk isn't critical to performance in 

> gobmk though.


Is loading/storing a pair as cheap as loading/storing a single register?
In that case you could shrink-wrap per pair of registers instead.


Segher
Kyrill Tkachov Nov. 17, 2016, 2:55 p.m. UTC | #3
On 17/11/16 14:44, Segher Boessenkool wrote:
> Hi Kyrill,

>

> On Thu, Nov 17, 2016 at 02:22:08PM +0000, Kyrill Tkachov wrote:

>>>>>>>> I ran SPEC2006 on a Cortex-A72. Overall scores were neutral but there

>>>>>>>> were

>>>>>>>> some interesting swings.

>>>>>>>> 458.sjeng     +1.45%

>>>>>>>> 471.omnetpp   +2.19%

>>>>>>>> 445.gobmk     -2.01%

>>>>>>>>

>>>>>>>> On SPECFP:

>>>>>>>> 453.povray    +7.00%

>> After looking at the gobmk performance with performance counters it looks

>> like more icache pressure.

>> I see an increase in misses.

>> This looks to me like an effect of code size increase, though it is not

>> that large an increase (0.4% with SWS).

> Right.  I don't see how to improve on this (but see below); ideas welcome :-)

>

>> Branch mispredicts also go up a bit but not as much as icache misses.

> I don't see that happening -- for some testcases we get unlucky and have

> more branch predictor aliasing, and for some we have less, it's pretty

> random.  Some testcases are really sensitive to this.


Right, I don't think it's the branch prediction at fault in this case,
rather the above icache stuff.

>

>> I don't think there's anything we can do here, or at least that this patch

>> can do about it.

>> Overall, there's a slight improvement in SPECINT, even with the gobmk

>> regression and a slightly larger improvement

>> on SPECFP due to povray.

> And that is for only the "normal" GPRs, not LR or FP yet, right?


This patch does implement FP registers wrapping as well but not LR.
Though I remember seeing the improvement even when only GPRs were wrapped
in an earlier version of the patch.

>> Segher, one curious artifact I spotted while looking at codegen differences

>> in gobmk was a case where we fail

>> to emit load-pairs as effectively in the epilogue and its preceeding basic

>> block.

>> So before we had this epilogue:

>> .L43:

>>      ldp    x21, x22, [sp, 16]

>>      ldp    x23, x24, [sp, 32]

>>      ldp    x25, x26, [sp, 48]

>>      ldp    x27, x28, [sp, 64]

>>      ldr    x30, [sp, 80]

>>      ldp    x19, x20, [sp], 112

>>      ret

>>

>> and I see this becoming (among numerous other changes in the function):

>>

>> .L69:

>>      ldp    x21, x22, [sp, 16]

>>      ldr    x24, [sp, 40]

>> .L43:

>>      ldp    x25, x26, [sp, 48]

>>      ldp    x27, x28, [sp, 64]

>>      ldr    x23, [sp, 32]

>>      ldr    x30, [sp, 80]

>>      ldp    x19, x20, [sp], 112

>>      ret

>>

>> So this is better in the cases where we jump straight into .L43 because we

>> load fewer registers

>> but worse when we jump to or fallthrough to .L69 because x23 and x24 are

>> now restored using two loads

>> rather than a single load-pair. This hunk isn't critical to performance in

>> gobmk though.

> Is loading/storing a pair as cheap as loading/storing a single register?

> In that case you could shrink-wrap per pair of registers instead.


I suppose it can vary by microarchitecture. For the purposes of codegen I'd say
it's more expensive than load/storing a single register (as there's more memory bandwidth required after all)
but cheaper than two separate loads stores (alignment quirks notwithstanding).
Interesting idea. That could help with code size too. I'll try it out.
Thanks,
Kyrill

>

> Segher
Segher Boessenkool Nov. 17, 2016, 3:01 p.m. UTC | #4
On Thu, Nov 17, 2016 at 02:55:20PM +0000, Kyrill Tkachov wrote:
> >>Overall, there's a slight improvement in SPECINT, even with the gobmk

> >>regression and a slightly larger improvement

> >>on SPECFP due to povray.

> >And that is for only the "normal" GPRs, not LR or FP yet, right?

> 

> This patch does implement FP registers wrapping as well but not LR.


I meant frame pointer, not floating point :-)


Segher
Kyrill Tkachov Nov. 17, 2016, 3:06 p.m. UTC | #5
On 17/11/16 15:01, Segher Boessenkool wrote:
> On Thu, Nov 17, 2016 at 02:55:20PM +0000, Kyrill Tkachov wrote:

>>>> Overall, there's a slight improvement in SPECINT, even with the gobmk

>>>> regression and a slightly larger improvement

>>>> on SPECFP due to povray.

>>> And that is for only the "normal" GPRs, not LR or FP yet, right?

>> This patch does implement FP registers wrapping as well but not LR.

> I meant frame pointer, not floating point :-)


Ah :) HARD_FRAME_POINTER is wrapped with -fomit-frame-pointer but we don't
touch the stack pointer (SP) in any case.

Kyrill

>

> Segher
Kyrill Tkachov Nov. 17, 2016, 4:50 p.m. UTC | #6
On 17/11/16 14:55, Kyrill Tkachov wrote:
>

> On 17/11/16 14:44, Segher Boessenkool wrote:

>> Hi Kyrill,

>>

>> On Thu, Nov 17, 2016 at 02:22:08PM +0000, Kyrill Tkachov wrote:

>>>>>>>>> I ran SPEC2006 on a Cortex-A72. Overall scores were neutral but there

>>>>>>>>> were

>>>>>>>>> some interesting swings.

>>>>>>>>> 458.sjeng     +1.45%

>>>>>>>>> 471.omnetpp   +2.19%

>>>>>>>>> 445.gobmk     -2.01%

>>>>>>>>>

>>>>>>>>> On SPECFP:

>>>>>>>>> 453.povray    +7.00%

>>> After looking at the gobmk performance with performance counters it looks

>>> like more icache pressure.

>>> I see an increase in misses.

>>> This looks to me like an effect of code size increase, though it is not

>>> that large an increase (0.4% with SWS).

>> Right.  I don't see how to improve on this (but see below); ideas welcome :-)

>>

>>> Branch mispredicts also go up a bit but not as much as icache misses.

>> I don't see that happening -- for some testcases we get unlucky and have

>> more branch predictor aliasing, and for some we have less, it's pretty

>> random.  Some testcases are really sensitive to this.

>

> Right, I don't think it's the branch prediction at fault in this case,

> rather the above icache stuff.

>

>>

>>> I don't think there's anything we can do here, or at least that this patch

>>> can do about it.

>>> Overall, there's a slight improvement in SPECINT, even with the gobmk

>>> regression and a slightly larger improvement

>>> on SPECFP due to povray.

>> And that is for only the "normal" GPRs, not LR or FP yet, right?

>

> This patch does implement FP registers wrapping as well but not LR.

> Though I remember seeing the improvement even when only GPRs were wrapped

> in an earlier version of the patch.

>

>>> Segher, one curious artifact I spotted while looking at codegen differences

>>> in gobmk was a case where we fail

>>> to emit load-pairs as effectively in the epilogue and its preceeding basic

>>> block.

>>> So before we had this epilogue:

>>> .L43:

>>>      ldp    x21, x22, [sp, 16]

>>>      ldp    x23, x24, [sp, 32]

>>>      ldp    x25, x26, [sp, 48]

>>>      ldp    x27, x28, [sp, 64]

>>>      ldr    x30, [sp, 80]

>>>      ldp    x19, x20, [sp], 112

>>>      ret

>>>

>>> and I see this becoming (among numerous other changes in the function):

>>>

>>> .L69:

>>>      ldp    x21, x22, [sp, 16]

>>>      ldr    x24, [sp, 40]

>>> .L43:

>>>      ldp    x25, x26, [sp, 48]

>>>      ldp    x27, x28, [sp, 64]

>>>      ldr    x23, [sp, 32]

>>>      ldr    x30, [sp, 80]

>>>      ldp    x19, x20, [sp], 112

>>>      ret

>>>

>>> So this is better in the cases where we jump straight into .L43 because we

>>> load fewer registers

>>> but worse when we jump to or fallthrough to .L69 because x23 and x24 are

>>> now restored using two loads

>>> rather than a single load-pair. This hunk isn't critical to performance in

>>> gobmk though.

>> Is loading/storing a pair as cheap as loading/storing a single register?

>> In that case you could shrink-wrap per pair of registers instead.

>

> I suppose it can vary by microarchitecture. For the purposes of codegen I'd say

> it's more expensive than load/storing a single register (as there's more memory bandwidth required after all)

> but cheaper than two separate loads stores (alignment quirks notwithstanding).

> Interesting idea. That could help with code size too. I'll try it out.


I'm encountering some difficulties implementing this idea.
I want to still keep the per-register structures across the hooks but basically restrict the number
of components in a basic block to an even number of FPRs and GPRs. I tried doing this in COMPONENTS_FOR_BB
but apparently this ended up not saving/restoring some of the registers at all because the components that were
"filtered out" that way still made their way to the bitmap passed into SET_HANDLED_COMPONENTS and so the normal
prologue/epilogue didn't end up saving and restoring them.

I don't want to do it in GET_SEPARATE_COMPONENTS as that doesn't see each basic block.
Is this something DISQUALIFY_COMPONENTS could be used for?

Thanks,
Kyrill

> Thanks,

> Kyrill

>

>>

>> Segher

>
Segher Boessenkool Nov. 17, 2016, 5:45 p.m. UTC | #7
On Thu, Nov 17, 2016 at 04:50:46PM +0000, Kyrill Tkachov wrote:
> >>Is loading/storing a pair as cheap as loading/storing a single register?

> >>In that case you could shrink-wrap per pair of registers instead.

> >

> >I suppose it can vary by microarchitecture. For the purposes of codegen 

> >I'd say

> >it's more expensive than load/storing a single register (as there's more 

> >memory bandwidth required after all)

> >but cheaper than two separate loads stores (alignment quirks 

> >notwithstanding).

> >Interesting idea. That could help with code size too. I'll try it out.

> 

> I'm encountering some difficulties implementing this idea.

> I want to still keep the per-register structures across the hooks but 

> basically restrict the number

> of components in a basic block to an even number of FPRs and GPRs. I tried 

> doing this in COMPONENTS_FOR_BB


So your COMPONENTS_FOR_BB returns both components in a pair whenever one
of those is needed?  That should work afaics.

> but apparently this ended up not saving/restoring some of the registers at 

> all because the components that were

> "filtered out" that way still made their way to the bitmap passed into 

> SET_HANDLED_COMPONENTS and so the normal

> prologue/epilogue didn't end up saving and restoring them.


I am not sure what this means?  "filtered out"?


Segher
Kyrill Tkachov Nov. 18, 2016, 9:29 a.m. UTC | #8
On 17/11/16 17:45, Segher Boessenkool wrote:
> On Thu, Nov 17, 2016 at 04:50:46PM +0000, Kyrill Tkachov wrote:

>>>> Is loading/storing a pair as cheap as loading/storing a single register?

>>>> In that case you could shrink-wrap per pair of registers instead.

>>> I suppose it can vary by microarchitecture. For the purposes of codegen

>>> I'd say

>>> it's more expensive than load/storing a single register (as there's more

>>> memory bandwidth required after all)

>>> but cheaper than two separate loads stores (alignment quirks

>>> notwithstanding).

>>> Interesting idea. That could help with code size too. I'll try it out.

>> I'm encountering some difficulties implementing this idea.

>> I want to still keep the per-register structures across the hooks but

>> basically restrict the number

>> of components in a basic block to an even number of FPRs and GPRs. I tried

>> doing this in COMPONENTS_FOR_BB

> So your COMPONENTS_FOR_BB returns both components in a pair whenever one

> of those is needed?  That should work afaics.

>


I mean I still want to have one component per register and since
emit_{prologue,epilogue}_components knows how to form pairs from the
components passed down to it I just need to restrict the number of
components in any particular basic block to an even number.
So say a function can wrap 5 registers: x22,x23,x24,x25,x26.
I want get_separate_components to return 5 components since in that hook
we don't know how these registers are distributed across each basic block.
components_for_bb has that information.
In components_for_bb I want to restrict the components for a basic block to
an even number, so if normally all 5 registers would be valid for wrapping
in that bb I'd only choose 4 so I could form 2 pairs. But selecting only 4
of the 5 registers, say only x22,x23,x24,x25 leads to x26 not being saved
or restored at all, even during the normal prologue and epilogue because
x26 was marked as a component in components_for_bb and therefore omitted from
the prologue and epilogue.
So I'm thinking x26 should be removed from the wrappable components of
a basic block by disqualify_components. I'm trying that approach now.

Thanks,
Kyrill

>> but apparently this ended up not saving/restoring some of the registers at

>> all because the components that were

>> "filtered out" that way still made their way to the bitmap passed into

>> SET_HANDLED_COMPONENTS and so the normal

>> prologue/epilogue didn't end up saving and restoring them.

> I am not sure what this means?  "filtered out"?

>

>

> Segher
Segher Boessenkool Nov. 18, 2016, 12:50 p.m. UTC | #9
On Fri, Nov 18, 2016 at 09:29:13AM +0000, Kyrill Tkachov wrote:
> >So your COMPONENTS_FOR_BB returns both components in a pair whenever one

> >of those is needed?  That should work afaics.

> 

> I mean I still want to have one component per register and since

> emit_{prologue,epilogue}_components knows how to form pairs from the

> components passed down to it I just need to restrict the number of

> components in any particular basic block to an even number.

> So say a function can wrap 5 registers: x22,x23,x24,x25,x26.

> I want get_separate_components to return 5 components since in that hook

> we don't know how these registers are distributed across each basic block.

> components_for_bb has that information.

> In components_for_bb I want to restrict the components for a basic block to

> an even number, so if normally all 5 registers would be valid for wrapping

> in that bb I'd only choose 4 so I could form 2 pairs. But selecting only 4

> of the 5 registers, say only x22,x23,x24,x25 leads to x26 not being saved

> or restored at all, even during the normal prologue and epilogue because

> x26 was marked as a component in components_for_bb and therefore omitted 

> from

> the prologue and epilogue.

> So I'm thinking x26 should be removed from the wrappable components of

> a basic block by disqualify_components. I'm trying that approach now.


My suggestion was, in components_for_bb, whenever you mark x22 as needed
you also mark x23 as needed, and whenever you mark x23 as needed you also
mark x22.  I think this is a lot simpler?


Segher
Kyrill Tkachov Nov. 22, 2016, 4:46 p.m. UTC | #10
On 18/11/16 12:50, Segher Boessenkool wrote:
> On Fri, Nov 18, 2016 at 09:29:13AM +0000, Kyrill Tkachov wrote:

>>> So your COMPONENTS_FOR_BB returns both components in a pair whenever one

>>> of those is needed?  That should work afaics.

>> I mean I still want to have one component per register and since

>> emit_{prologue,epilogue}_components knows how to form pairs from the

>> components passed down to it I just need to restrict the number of

>> components in any particular basic block to an even number.

>> So say a function can wrap 5 registers: x22,x23,x24,x25,x26.

>> I want get_separate_components to return 5 components since in that hook

>> we don't know how these registers are distributed across each basic block.

>> components_for_bb has that information.

>> In components_for_bb I want to restrict the components for a basic block to

>> an even number, so if normally all 5 registers would be valid for wrapping

>> in that bb I'd only choose 4 so I could form 2 pairs. But selecting only 4

>> of the 5 registers, say only x22,x23,x24,x25 leads to x26 not being saved

>> or restored at all, even during the normal prologue and epilogue because

>> x26 was marked as a component in components_for_bb and therefore omitted

>> from

>> the prologue and epilogue.

>> So I'm thinking x26 should be removed from the wrappable components of

>> a basic block by disqualify_components. I'm trying that approach now.

> My suggestion was, in components_for_bb, whenever you mark x22 as needed

> you also mark x23 as needed, and whenever you mark x23 as needed you also

> mark x22.  I think this is a lot simpler?


But then we'd have cases where we're saving and restoring x23
even when it's not necessary.
In any case, I tried it out and it didn't fix the gobmk issue, though it did reduce the code
size increase somewhat.

With the patch already posted at [1] the net result is still positive on
both SPECINT and SPECFP.

I also ran the numbers on a Cortex-A57. The changes are less pronounced
with SPECINT being neutral (gobmk shows only a 0.8% regression) and SPECFP
having a small improvement, due to povray improving by 2.9%.

Thanks,
Kyrill

[1] https://gcc.gnu.org/ml/gcc-patches/2016-11/msg01352.html

>

> Segher
James Greenhalgh Nov. 29, 2016, 10:57 a.m. UTC | #11
On Mon, Nov 14, 2016 at 02:25:28PM +0000, Kyrill Tkachov wrote:
> 

> On 11/11/16 15:31, Kyrill Tkachov wrote:

> >

> >On 11/11/16 10:17, Kyrill Tkachov wrote:

> >>

> >>On 10/11/16 23:39, Segher Boessenkool wrote:

> >>>On Thu, Nov 10, 2016 at 02:42:24PM -0800, Andrew Pinski wrote:

> >>>>On Thu, Nov 10, 2016 at 6:25 AM, Kyrill Tkachov

> >>>>>I ran SPEC2006 on a Cortex-A72. Overall scores were neutral but there were

> >>>>>some interesting swings.

> >>>>>458.sjeng     +1.45%

> >>>>>471.omnetpp   +2.19%

> >>>>>445.gobmk     -2.01%

> >>>>>

> >>>>>On SPECFP:

> >>>>>453.povray    +7.00%

> >>>>

> >>>>Wow, this looks really good.  Thank you for implementing this.  If I

> >>>>get some time I am going to try it out on other processors than A72

> >>>>but I doubt I have time any time soon.

> >>>I'd love to hear what causes the slowdown for gobmk as well, btw.

> >>

> >>I haven't yet gotten a direct answer for that (through performance analysis

> >>tools) but I have noticed that load/store pairs are not generated as

> >>aggressively as I hoped.  They are being merged by the sched fusion pass

> >>and peepholes (which runs after this) but it still misses cases. I've

> >>hacked the SWS hooks to generate pairs explicitly and that increases the

> >>number of pairs and helps code size to boot. It complicates the logic of

> >>the hooks a bit but not too much.

> >>

> >>I'll make those changes and re-benchmark, hopefully that

> >>will help performance.

> >>

> >

> >And here's a version that explicitly emits pairs. I've looked at assembly

> >codegen on SPEC2006 and it generates quite a few more LDP/STP pairs than the

> >original version.  I kicked off benchmarks over the weekend to see the

> >effect.  Andrew, if you want to try it out (more benchmarking and testing

> >always welcome) this is the one to try.

> >

> 

> And I discovered over the weekend that gamess and wrf have validation errors.

> This version runs correctly.  SPECINT results were fine though and there is

> even a small overall gain due to sjeng and omnetpp. However, gobmk still has

> the regression.  I'll rerun SPECFP with this patch (it's really just a small

> bugfix over the previous version) and get on with analysing gobmk.


I have some comments in line, most of which are about hardcoding the
maximum register number, but at a high level this looks good to me.

Thanks,
James

> 2016-11-11  Kyrylo Tkachov  <kyrylo.tkachov@arm.com>

> 

>     * config/aarch64/aarch64.h (machine_function): Add

>     reg_is_wrapped_separately field.

>     * config/aarch64/aarch64.c (emit_set_insn): Change return type to

>     rtx_insn *.

>     (aarch64_save_callee_saves): Don't save registers that are wrapped

>     separately.

>     (aarch64_restore_callee_saves): Don't restore registers that are

>     wrapped separately.

>     (offset_9bit_signed_unscaled_p, offset_12bit_unsigned_scaled_p,

>     aarch64_offset_7bit_signed_scaled_p): Move earlier in the file.

>     (aarch64_get_separate_components): New function.

>     (aarch64_get_next_set_bit): Likewise.

>     (aarch64_components_for_bb): Likewise.

>     (aarch64_disqualify_components): Likewise.

>     (aarch64_emit_prologue_components): Likewise.

>     (aarch64_emit_epilogue_components): Likewise.

>     (aarch64_set_handled_components): Likewise.

>     (TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS,

>     TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB,

>     TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS,

>     TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS,

>     TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS,

>     TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS): Define.

> 


> commit 06ac3c30d8aa38781ee9019e60a5fcf727b85231

> Author: Kyrylo Tkachov <kyrylo.tkachov@arm.com>

> Date:   Tue Oct 11 09:25:54 2016 +0100

> 

>     [AArch64] Separate shrink wrapping hooks implementation

> 

> diff --git a/gcc/config/aarch64/aarch64.c b/gcc/config/aarch64/aarch64.c

> index 325e725..2d33ef6 100644

> --- a/gcc/config/aarch64/aarch64.c

> +++ b/gcc/config/aarch64/aarch64.c

> @@ -1138,7 +1138,7 @@ aarch64_is_extend_from_extract (machine_mode mode, rtx mult_imm,

>  

>  /* Emit an insn that's a simple single-set.  Both the operands must be

>     known to be valid.  */

> -inline static rtx

> +inline static rtx_insn *

>  emit_set_insn (rtx x, rtx y)

>  {

>    return emit_insn (gen_rtx_SET (x, y));

> @@ -3135,6 +3135,9 @@ aarch64_save_callee_saves (machine_mode mode, HOST_WIDE_INT start_offset,

>  	      || regno == cfun->machine->frame.wb_candidate2))

>  	continue;

>  

> +      if (cfun->machine->reg_is_wrapped_separately[regno])

> +       continue;

> +

>        reg = gen_rtx_REG (mode, regno);

>        offset = start_offset + cfun->machine->frame.reg_offset[regno];

>        mem = gen_mem_ref (mode, plus_constant (Pmode, stack_pointer_rtx,

> @@ -3143,6 +3146,7 @@ aarch64_save_callee_saves (machine_mode mode, HOST_WIDE_INT start_offset,

>        regno2 = aarch64_next_callee_save (regno + 1, limit);

>  

>        if (regno2 <= limit

> +	  && !cfun->machine->reg_is_wrapped_separately[regno2]

>  	  && ((cfun->machine->frame.reg_offset[regno] + UNITS_PER_WORD)

>  	      == cfun->machine->frame.reg_offset[regno2]))

>  

> @@ -3191,6 +3195,9 @@ aarch64_restore_callee_saves (machine_mode mode,

>         regno <= limit;

>         regno = aarch64_next_callee_save (regno + 1, limit))

>      {

> +      if (cfun->machine->reg_is_wrapped_separately[regno])

> +       continue;

> +

>        rtx reg, mem;

>  

>        if (skip_wb

> @@ -3205,6 +3212,7 @@ aarch64_restore_callee_saves (machine_mode mode,

>        regno2 = aarch64_next_callee_save (regno + 1, limit);

>  

>        if (regno2 <= limit

> +	  && !cfun->machine->reg_is_wrapped_separately[regno2]

>  	  && ((cfun->machine->frame.reg_offset[regno] + UNITS_PER_WORD)

>  	      == cfun->machine->frame.reg_offset[regno2]))

>  	{

> @@ -3224,6 +3232,272 @@ aarch64_restore_callee_saves (machine_mode mode,

>      }

>  }

>  

> +static inline bool

> +offset_9bit_signed_unscaled_p (machine_mode mode ATTRIBUTE_UNUSED,

> +			       HOST_WIDE_INT offset)

> +{

> +  return offset >= -256 && offset < 256;


A follow-up changing this to use IN_RANGE would seem correct.

> +}

> +

> +static inline bool

> +offset_12bit_unsigned_scaled_p (machine_mode mode, HOST_WIDE_INT offset)

> +{

> +  return (offset >= 0

> +	  && offset < 4096 * GET_MODE_SIZE (mode)

> +	  && offset % GET_MODE_SIZE (mode) == 0);

> +}

> +

> +bool

> +aarch64_offset_7bit_signed_scaled_p (machine_mode mode, HOST_WIDE_INT offset)

> +{

> +  return (offset >= -64 * GET_MODE_SIZE (mode)

> +	  && offset < 64 * GET_MODE_SIZE (mode)

> +	  && offset % GET_MODE_SIZE (mode) == 0);


And certainly here, IN_RANGE would be neater.

> +}

> +

> +/* Implement TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS.  */

> +

> +static sbitmap

> +aarch64_get_separate_components (void)

> +{

> +  aarch64_layout_frame ();

> +

> +  sbitmap components = sbitmap_alloc (V31_REGNUM + 1);

> +  bitmap_clear (components);

> +

> +  /* The registers we need saved to the frame.  */

> +  for (unsigned regno = R0_REGNUM; regno <= V31_REGNUM; regno++)

> +    if (aarch64_register_saved_on_entry (regno))

> +      {

> +	HOST_WIDE_INT offset = cfun->machine->frame.reg_offset[regno];

> +	if (!frame_pointer_needed)

> +	  offset += cfun->machine->frame.frame_size

> +		    - cfun->machine->frame.hard_fp_offset;

> +	/* Check that we can access the stack slot of the register with one

> +	   direct load with no adjustments needed.  */

> +	if (offset_12bit_unsigned_scaled_p (DImode, offset))

> +	  bitmap_set_bit (components, regno);

> +      }

> +

> +  /* Don't mess with the hard frame pointer.  */

> +  if (frame_pointer_needed)

> +    bitmap_clear_bit (components, HARD_FRAME_POINTER_REGNUM);

> +

> +  unsigned reg1 = cfun->machine->frame.wb_candidate1;

> +  unsigned reg2 = cfun->machine->frame.wb_candidate2;

> +  /* If aarch64_layout_frame has chosen registers to store/restore with

> +     writeback don't interfere with them to avoid having to output explicit

> +     stack adjustment instructions.  */

> +  if (reg2 != INVALID_REGNUM)

> +    bitmap_clear_bit (components, reg2);

> +  if (reg1 != INVALID_REGNUM)

> +    bitmap_clear_bit (components, reg1);

> +

> +  bitmap_clear_bit (components, LR_REGNUM);

> +  bitmap_clear_bit (components, SP_REGNUM);

> +

> +  return components;

> +}

> +

> +/* Implement TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB.  */

> +

> +static sbitmap

> +aarch64_components_for_bb (basic_block bb)

> +{

> +  bitmap in = DF_LIVE_IN (bb);

> +  bitmap gen = &DF_LIVE_BB_INFO (bb)->gen;

> +  bitmap kill = &DF_LIVE_BB_INFO (bb)->kill;

> +

> +  sbitmap components = sbitmap_alloc (V31_REGNUM + 1);

> +  bitmap_clear (components);

> +

> +  /* GPRs are used in a bb if they are in the IN, GEN, or KILL sets.  */

> +  for (unsigned regno = R0_REGNUM; regno <= V31_REGNUM; regno++)


The use of R0_REGNUM and V31_REGNUM scare me a little bit, as we're hardcoding
where the end of the register file is (does this, for example, fall apart
with the SVE work that was recently posted). Something like a
LAST_HARDREG_NUM might work?

> +    if ((!call_used_regs[regno])

> +       && (bitmap_bit_p (in, regno)

> +	   || bitmap_bit_p (gen, regno)

> +	   || bitmap_bit_p (kill, regno)))

> +	  bitmap_set_bit (components, regno);

> +

> +  return components;

> +}

> +

> +/* Implement TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS.

> +   Nothing to do for aarch64.  */

> +

> +static void

> +aarch64_disqualify_components (sbitmap, edge, sbitmap, bool)

> +{

> +}


Is there no default "do nothing" hook for this?

> +

> +/* Return the next set bit in BMP from START onwards.  Return the total number

> +   of bits in BMP if no set bit is found at or after START.  */

> +

> +static unsigned int

> +aarch64_get_next_set_bit (sbitmap bmp, unsigned int start)

> +{

> +  unsigned int nbits = SBITMAP_SIZE (bmp);

> +  if (start == nbits)

> +    return start;

> +

> +  gcc_assert (start < nbits);

> +  for (unsigned int i = start; i < nbits; i++)

> +    if (bitmap_bit_p (bmp, i))

> +      return i;

> +

> +  return nbits;

> +}

> +

> +/* Implement TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS.  */

> +

> +static void

> +aarch64_emit_prologue_components (sbitmap components)

> +{

> +  rtx ptr_reg = gen_rtx_REG (Pmode, frame_pointer_needed

> +			     ? HARD_FRAME_POINTER_REGNUM

> +			     : STACK_POINTER_REGNUM);

> +

> +  unsigned total_bits = SBITMAP_SIZE (components);


Would this be clearer called last_regno ?

> +  unsigned regno = aarch64_get_next_set_bit (components, R0_REGNUM);

> +  rtx_insn *insn = NULL;

> +

> +  while (regno != total_bits)

> +    {

> +      machine_mode mode = GP_REGNUM_P (regno) ? DImode : DFmode;


Comment about why this can be DFmode rather than some 128-bit mode might
be useful here.

> +      rtx reg = gen_rtx_REG (mode, regno);

> +      HOST_WIDE_INT offset = cfun->machine->frame.reg_offset[regno];

> +      if (!frame_pointer_needed)

> +	offset += cfun->machine->frame.frame_size

> +		  - cfun->machine->frame.hard_fp_offset;

> +      rtx addr = plus_constant (Pmode, ptr_reg, offset);

> +      rtx mem = gen_frame_mem (mode, addr);

> +

> +      rtx set = gen_rtx_SET (mem, reg);

> +      unsigned regno2 = aarch64_get_next_set_bit (components, regno + 1);

> +      /* No more registers to save after REGNO.

> +	 Emit a single save and exit.  */

> +      if (regno2 == total_bits)

> +	{

> +	  insn = emit_insn (set);

> +	  RTX_FRAME_RELATED_P (insn) = 1;

> +	  add_reg_note (insn, REG_CFA_OFFSET, copy_rtx (set));

> +	  break;

> +	}

> +

> +      HOST_WIDE_INT offset2 = cfun->machine->frame.reg_offset[regno2];

> +      /* The next register is not of the same class or its offset is not

> +	 mergeable with the current one into a pair.  */

> +      if (!satisfies_constraint_Ump (mem)

> +	  || GP_REGNUM_P (regno) != GP_REGNUM_P (regno2)

> +	  || (offset2 - cfun->machine->frame.reg_offset[regno])

> +		!= GET_MODE_SIZE (DImode))

> +	{

> +	  insn = emit_insn (set);

> +	  RTX_FRAME_RELATED_P (insn) = 1;

> +	  add_reg_note (insn, REG_CFA_OFFSET, copy_rtx (set));

> +

> +	  regno = regno2;

> +	  continue;

> +	}

> +

> +      /* REGNO2 can be stored in a pair with REGNO.  */

> +      rtx reg2 = gen_rtx_REG (mode, regno2);

> +      if (!frame_pointer_needed)

> +	offset2 += cfun->machine->frame.frame_size

> +		  - cfun->machine->frame.hard_fp_offset;

> +      rtx addr2 = plus_constant (Pmode, ptr_reg, offset2);

> +      rtx mem2 = gen_frame_mem (mode, addr2);

> +      rtx set2 = gen_rtx_SET (mem2, reg2);

> +

> +      insn = emit_insn (aarch64_gen_store_pair (mode, mem, reg, mem2, reg2));

> +      RTX_FRAME_RELATED_P (insn) = 1;

> +      add_reg_note (insn, REG_CFA_OFFSET, set);

> +      add_reg_note (insn, REG_CFA_OFFSET, set2);

> +

> +      regno = aarch64_get_next_set_bit (components, regno2 + 1);

> +    }

> +

> +}

> +

> +/* Implement TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS.  */

> +

> +static void

> +aarch64_emit_epilogue_components (sbitmap components)


Given the similarity of the logic, is there any way you can refactor this
with the prologue code above?

> +{

> +

> +  rtx ptr_reg = gen_rtx_REG (Pmode, frame_pointer_needed

> +			     ? HARD_FRAME_POINTER_REGNUM

> +			     : STACK_POINTER_REGNUM);

> +  unsigned total_bits = SBITMAP_SIZE (components);

> +  unsigned regno = aarch64_get_next_set_bit (components, R0_REGNUM);

> +  rtx_insn *insn = NULL;

> +

> +  while (regno != total_bits)

> +    {

> +      machine_mode mode = GP_REGNUM_P (regno) ? DImode : DFmode;

> +      rtx reg = gen_rtx_REG (mode, regno);

> +      HOST_WIDE_INT offset = cfun->machine->frame.reg_offset[regno];

> +      if (!frame_pointer_needed)

> +	offset += cfun->machine->frame.frame_size

> +		  - cfun->machine->frame.hard_fp_offset;

> +      rtx addr = plus_constant (Pmode, ptr_reg, offset);

> +      rtx mem = gen_frame_mem (mode, addr);

> +

> +      unsigned regno2 = aarch64_get_next_set_bit (components, regno + 1);

> +      /* No more registers after REGNO to restore.

> +	 Emit a single restore and exit.  */

> +      if (regno2 == total_bits)

> +	{

> +	  insn = emit_move_insn (reg, mem);

> +	  RTX_FRAME_RELATED_P (insn) = 1;

> +	  add_reg_note (insn, REG_CFA_RESTORE, reg);

> +	  break;

> +	}

> +

> +      HOST_WIDE_INT offset2 = cfun->machine->frame.reg_offset[regno2];

> +      /* The next register is not of the same class or its offset is not

> +	 mergeable with the current one into a pair or the offset doesn't fit

> +	 for a load pair.  Emit a single restore and continue from REGNO2.  */

> +      if (!satisfies_constraint_Ump (mem)

> +	  || GP_REGNUM_P (regno) != GP_REGNUM_P (regno2)

> +	  || (offset2 - cfun->machine->frame.reg_offset[regno])

> +		!= GET_MODE_SIZE (DImode))

> +	{

> +	  insn = emit_move_insn (reg, mem);

> +	  RTX_FRAME_RELATED_P (insn) = 1;

> +	  add_reg_note (insn, REG_CFA_RESTORE, reg);

> +

> +	  regno = regno2;

> +	  continue;

> +	}

> +

> +      /* REGNO2 can be loaded in a pair with REGNO.  */

> +      rtx reg2 = gen_rtx_REG (mode, regno2);

> +      if (!frame_pointer_needed)

> +	offset2 += cfun->machine->frame.frame_size

> +		  - cfun->machine->frame.hard_fp_offset;

> +      rtx addr2 = plus_constant (Pmode, ptr_reg, offset2);

> +      rtx mem2 = gen_frame_mem (mode, addr2);

> +

> +      insn = emit_insn (aarch64_gen_load_pair (mode, reg, mem, reg2, mem2));

> +      RTX_FRAME_RELATED_P (insn) = 1;

> +      add_reg_note (insn, REG_CFA_RESTORE, reg);

> +      add_reg_note (insn, REG_CFA_RESTORE, reg2);

> +

> +      regno = aarch64_get_next_set_bit (components, regno2 + 1);

> +    }

> +}

> +

> +/* Implement TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS.  */

> +

> +static void

> +aarch64_set_handled_components (sbitmap components)

> +{

> +  for (unsigned regno = R0_REGNUM; regno <= V31_REGNUM; regno++)

> +    if (bitmap_bit_p (components, regno))

> +      cfun->machine->reg_is_wrapped_separately[regno] = true;

> +}

> +

>  /* AArch64 stack frames generated by this compiler look like:

>  

>  	+-------------------------------+

> @@ -3944,29 +4218,6 @@ aarch64_classify_index (struct aarch64_address_info *info, rtx x,

>    return false;

>  }

>  

> -bool

> -aarch64_offset_7bit_signed_scaled_p (machine_mode mode, HOST_WIDE_INT offset)

> -{

> -  return (offset >= -64 * GET_MODE_SIZE (mode)

> -	  && offset < 64 * GET_MODE_SIZE (mode)

> -	  && offset % GET_MODE_SIZE (mode) == 0);

> -}

> -

> -static inline bool

> -offset_9bit_signed_unscaled_p (machine_mode mode ATTRIBUTE_UNUSED,

> -			       HOST_WIDE_INT offset)

> -{

> -  return offset >= -256 && offset < 256;

> -}

> -

> -static inline bool

> -offset_12bit_unsigned_scaled_p (machine_mode mode, HOST_WIDE_INT offset)

> -{

> -  return (offset >= 0

> -	  && offset < 4096 * GET_MODE_SIZE (mode)

> -	  && offset % GET_MODE_SIZE (mode) == 0);

> -}

> -

>  /* Return true if MODE is one of the modes for which we

>     support LDP/STP operations.  */

>  

> @@ -14452,6 +14703,30 @@ aarch64_optab_supported_p (int op, machine_mode mode1, machine_mode,

>  #define TARGET_SCHED_FIRST_CYCLE_MULTIPASS_DFA_LOOKAHEAD_GUARD \

>    aarch64_first_cycle_multipass_dfa_lookahead_guard

>  

> +#undef TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS

> +#define TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS \

> +  aarch64_get_separate_components

> +

> +#undef TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB

> +#define TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB \

> +  aarch64_components_for_bb

> +

> +#undef TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS

> +#define TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS \

> +  aarch64_disqualify_components

> +

> +#undef TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS

> +#define TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS \

> +  aarch64_emit_prologue_components

> +

> +#undef TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS

> +#define TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS \

> +  aarch64_emit_epilogue_components

> +

> +#undef TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS

> +#define TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS \

> +  aarch64_set_handled_components

> +

>  #undef TARGET_TRAMPOLINE_INIT

>  #define TARGET_TRAMPOLINE_INIT aarch64_trampoline_init

>  

> diff --git a/gcc/config/aarch64/aarch64.h b/gcc/config/aarch64/aarch64.h

> index 584ff5c..fb89e5a 100644

> --- a/gcc/config/aarch64/aarch64.h

> +++ b/gcc/config/aarch64/aarch64.h

> @@ -591,6 +591,8 @@ struct GTY (()) aarch64_frame

>  typedef struct GTY (()) machine_function

>  {

>    struct aarch64_frame frame;

> +  /* One entry for each GPR and FP register.  */

> +  bool reg_is_wrapped_separately[V31_REGNUM + 1];


Another hardcoded use of V31_REGNUM.

Thanks,
James
Kyrill Tkachov Nov. 29, 2016, 11:18 a.m. UTC | #12
Hi James,

On 29/11/16 10:57, James Greenhalgh wrote:
> On Mon, Nov 14, 2016 at 02:25:28PM +0000, Kyrill Tkachov wrote:

>> On 11/11/16 15:31, Kyrill Tkachov wrote:

>>> On 11/11/16 10:17, Kyrill Tkachov wrote:

>>>> On 10/11/16 23:39, Segher Boessenkool wrote:

>>>>> On Thu, Nov 10, 2016 at 02:42:24PM -0800, Andrew Pinski wrote:

>>>>>> On Thu, Nov 10, 2016 at 6:25 AM, Kyrill Tkachov

>>>>>>> I ran SPEC2006 on a Cortex-A72. Overall scores were neutral but there were

>>>>>>> some interesting swings.

>>>>>>> 458.sjeng     +1.45%

>>>>>>> 471.omnetpp   +2.19%

>>>>>>> 445.gobmk     -2.01%

>>>>>>>

>>>>>>> On SPECFP:

>>>>>>> 453.povray    +7.00%

>>>>>> Wow, this looks really good.  Thank you for implementing this.  If I

>>>>>> get some time I am going to try it out on other processors than A72

>>>>>> but I doubt I have time any time soon.

>>>>> I'd love to hear what causes the slowdown for gobmk as well, btw.

>>>> I haven't yet gotten a direct answer for that (through performance analysis

>>>> tools) but I have noticed that load/store pairs are not generated as

>>>> aggressively as I hoped.  They are being merged by the sched fusion pass

>>>> and peepholes (which runs after this) but it still misses cases. I've

>>>> hacked the SWS hooks to generate pairs explicitly and that increases the

>>>> number of pairs and helps code size to boot. It complicates the logic of

>>>> the hooks a bit but not too much.

>>>>

>>>> I'll make those changes and re-benchmark, hopefully that

>>>> will help performance.

>>>>

>>> And here's a version that explicitly emits pairs. I've looked at assembly

>>> codegen on SPEC2006 and it generates quite a few more LDP/STP pairs than the

>>> original version.  I kicked off benchmarks over the weekend to see the

>>> effect.  Andrew, if you want to try it out (more benchmarking and testing

>>> always welcome) this is the one to try.

>>>

>> And I discovered over the weekend that gamess and wrf have validation errors.

>> This version runs correctly.  SPECINT results were fine though and there is

>> even a small overall gain due to sjeng and omnetpp. However, gobmk still has

>> the regression.  I'll rerun SPECFP with this patch (it's really just a small

>> bugfix over the previous version) and get on with analysing gobmk.

> I have some comments in line, most of which are about hardcoding the

> maximum register number, but at a high level this looks good to me.


Thanks for having a look.
I'll respin with the comments addressed and I have a couple of comments inline.

Kyrill

> Thanks,

> James

>

>


<snip>

> +

> +/* Implement TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB.  */

> +

> +static sbitmap

> +aarch64_components_for_bb (basic_block bb)

> +{

> +  bitmap in = DF_LIVE_IN (bb);

> +  bitmap gen = &DF_LIVE_BB_INFO (bb)->gen;

> +  bitmap kill = &DF_LIVE_BB_INFO (bb)->kill;

> +

> +  sbitmap components = sbitmap_alloc (V31_REGNUM + 1);

> +  bitmap_clear (components);

> +

> +  /* GPRs are used in a bb if they are in the IN, GEN, or KILL sets.  */

> +  for (unsigned regno = R0_REGNUM; regno <= V31_REGNUM; regno++)

> The use of R0_REGNUM and V31_REGNUM scare me a little bit, as we're hardcoding

> where the end of the register file is (does this, for example, fall apart

> with the SVE work that was recently posted). Something like a

> LAST_HARDREG_NUM might work?

>


I think you mean FIRST_PSEUDO_REGISTER. AFAICS the compiler uses
a loop from 0 to FIRST_PSEUDO_REGISTER to go through the hard registers
in various places in the midend.
I'll change it to use that, though if the way to save/restore such new registers becomes
different from the current approach (i.e. perform a DI/DFmode memory op) the code in these
hooks will have to be updated anyway to take it into account.

>> +    if ((!call_used_regs[regno])

>> +       && (bitmap_bit_p (in, regno)

>> +	   || bitmap_bit_p (gen, regno)

>> +	   || bitmap_bit_p (kill, regno)))

>> +	  bitmap_set_bit (components, regno);

>> +

>> +  return components;

>> +}

>> +

>> +/* Implement TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS.

>> +   Nothing to do for aarch64.  */

>> +

>> +static void

>> +aarch64_disqualify_components (sbitmap, edge, sbitmap, bool)

>> +{

>> +}

> Is there no default "do nothing" hook for this?

>


I don't see one defined anywhere and the documentation for TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS
says that if it is defined, all other hooks in the group must be defined.

>> +

>> +/* Return the next set bit in BMP from START onwards.  Return the total number

>> +   of bits in BMP if no set bit is found at or after START.  */

>> +

>> +static unsigned int

>> +aarch64_get_next_set_bit (sbitmap bmp, unsigned int start)

>> +{

>> +  unsigned int nbits = SBITMAP_SIZE (bmp);

>> +  if (start == nbits)

>> +    return start;

>> +

>> +  gcc_assert (start < nbits);

>> +  for (unsigned int i = start; i < nbits; i++)

>> +    if (bitmap_bit_p (bmp, i))

>> +      return i;

>> +

>> +  return nbits;

>> +}

>> +

>> +/* Implement TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS.  */

>> +

>> +static void

>> +aarch64_emit_prologue_components (sbitmap components)

>> +{

>> +  rtx ptr_reg = gen_rtx_REG (Pmode, frame_pointer_needed

>> +			     ? HARD_FRAME_POINTER_REGNUM

>> +			     : STACK_POINTER_REGNUM);

>> +

>> +  unsigned total_bits = SBITMAP_SIZE (components);

> Would this be clearer called last_regno ?

>

>> +  unsigned regno = aarch64_get_next_set_bit (components, R0_REGNUM);

>> +  rtx_insn *insn = NULL;

>> +

>> +  while (regno != total_bits)

>> +    {

>> +      machine_mode mode = GP_REGNUM_P (regno) ? DImode : DFmode;

> Comment about why this can be DFmode rather than some 128-bit mode might

> be useful here.

>

>> +      rtx reg = gen_rtx_REG (mode, regno);

>> +      HOST_WIDE_INT offset = cfun->machine->frame.reg_offset[regno];

>> +      if (!frame_pointer_needed)

>> +	offset += cfun->machine->frame.frame_size

>> +		  - cfun->machine->frame.hard_fp_offset;

>> +      rtx addr = plus_constant (Pmode, ptr_reg, offset);

>> +      rtx mem = gen_frame_mem (mode, addr);

>> +

>> +      rtx set = gen_rtx_SET (mem, reg);

>> +      unsigned regno2 = aarch64_get_next_set_bit (components, regno + 1);

>> +      /* No more registers to save after REGNO.

>> +	 Emit a single save and exit.  */

>> +      if (regno2 == total_bits)

>> +	{

>> +	  insn = emit_insn (set);

>> +	  RTX_FRAME_RELATED_P (insn) = 1;

>> +	  add_reg_note (insn, REG_CFA_OFFSET, copy_rtx (set));

>> +	  break;

>> +	}

>> +

>> +      HOST_WIDE_INT offset2 = cfun->machine->frame.reg_offset[regno2];

>> +      /* The next register is not of the same class or its offset is not

>> +	 mergeable with the current one into a pair.  */

>> +      if (!satisfies_constraint_Ump (mem)

>> +	  || GP_REGNUM_P (regno) != GP_REGNUM_P (regno2)

>> +	  || (offset2 - cfun->machine->frame.reg_offset[regno])

>> +		!= GET_MODE_SIZE (DImode))

>> +	{

>> +	  insn = emit_insn (set);

>> +	  RTX_FRAME_RELATED_P (insn) = 1;

>> +	  add_reg_note (insn, REG_CFA_OFFSET, copy_rtx (set));

>> +

>> +	  regno = regno2;

>> +	  continue;

>> +	}

>> +

>> +      /* REGNO2 can be stored in a pair with REGNO.  */

>> +      rtx reg2 = gen_rtx_REG (mode, regno2);

>> +      if (!frame_pointer_needed)

>> +	offset2 += cfun->machine->frame.frame_size

>> +		  - cfun->machine->frame.hard_fp_offset;

>> +      rtx addr2 = plus_constant (Pmode, ptr_reg, offset2);

>> +      rtx mem2 = gen_frame_mem (mode, addr2);

>> +      rtx set2 = gen_rtx_SET (mem2, reg2);

>> +

>> +      insn = emit_insn (aarch64_gen_store_pair (mode, mem, reg, mem2, reg2));

>> +      RTX_FRAME_RELATED_P (insn) = 1;

>> +      add_reg_note (insn, REG_CFA_OFFSET, set);

>> +      add_reg_note (insn, REG_CFA_OFFSET, set2);

>> +

>> +      regno = aarch64_get_next_set_bit (components, regno2 + 1);

>> +    }

>> +

>> +}

>> +

>> +/* Implement TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS.  */

>> +

>> +static void

>> +aarch64_emit_epilogue_components (sbitmap components)

> Given the similarity of the logic, is there any way you can refactor this

> with the prologue code above?

>

>> +{

>> +

>> +  rtx ptr_reg = gen_rtx_REG (Pmode, frame_pointer_needed

>> +			     ? HARD_FRAME_POINTER_REGNUM

>> +			     : STACK_POINTER_REGNUM);

>> +  unsigned total_bits = SBITMAP_SIZE (components);

>> +  unsigned regno = aarch64_get_next_set_bit (components, R0_REGNUM);

>> +  rtx_insn *insn = NULL;

>> +

>> +  while (regno != total_bits)

>> +    {

>> +      machine_mode mode = GP_REGNUM_P (regno) ? DImode : DFmode;

>> +      rtx reg = gen_rtx_REG (mode, regno);

>> +      HOST_WIDE_INT offset = cfun->machine->frame.reg_offset[regno];

>> +      if (!frame_pointer_needed)

>> +	offset += cfun->machine->frame.frame_size

>> +		  - cfun->machine->frame.hard_fp_offset;

>> +      rtx addr = plus_constant (Pmode, ptr_reg, offset);

>> +      rtx mem = gen_frame_mem (mode, addr);

>> +

>> +      unsigned regno2 = aarch64_get_next_set_bit (components, regno + 1);

>> +      /* No more registers after REGNO to restore.

>> +	 Emit a single restore and exit.  */

>> +      if (regno2 == total_bits)

>> +	{

>> +	  insn = emit_move_insn (reg, mem);

>> +	  RTX_FRAME_RELATED_P (insn) = 1;

>> +	  add_reg_note (insn, REG_CFA_RESTORE, reg);

>> +	  break;

>> +	}

>> +

>> +      HOST_WIDE_INT offset2 = cfun->machine->frame.reg_offset[regno2];

>> +      /* The next register is not of the same class or its offset is not

>> +	 mergeable with the current one into a pair or the offset doesn't fit

>> +	 for a load pair.  Emit a single restore and continue from REGNO2.  */

>> +      if (!satisfies_constraint_Ump (mem)

>> +	  || GP_REGNUM_P (regno) != GP_REGNUM_P (regno2)

>> +	  || (offset2 - cfun->machine->frame.reg_offset[regno])

>> +		!= GET_MODE_SIZE (DImode))

>> +	{

>> +	  insn = emit_move_insn (reg, mem);

>> +	  RTX_FRAME_RELATED_P (insn) = 1;

>> +	  add_reg_note (insn, REG_CFA_RESTORE, reg);

>> +

>> +	  regno = regno2;

>> +	  continue;

>> +	}

>> +

>> +      /* REGNO2 can be loaded in a pair with REGNO.  */

>> +      rtx reg2 = gen_rtx_REG (mode, regno2);

>> +      if (!frame_pointer_needed)

>> +	offset2 += cfun->machine->frame.frame_size

>> +		  - cfun->machine->frame.hard_fp_offset;

>> +      rtx addr2 = plus_constant (Pmode, ptr_reg, offset2);

>> +      rtx mem2 = gen_frame_mem (mode, addr2);

>> +

>> +      insn = emit_insn (aarch64_gen_load_pair (mode, reg, mem, reg2, mem2));

>> +      RTX_FRAME_RELATED_P (insn) = 1;

>> +      add_reg_note (insn, REG_CFA_RESTORE, reg);

>> +      add_reg_note (insn, REG_CFA_RESTORE, reg2);

>> +

>> +      regno = aarch64_get_next_set_bit (components, regno2 + 1);

>> +    }

>> +}

>> +

>> +/* Implement TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS.  */

>> +

>> +static void

>> +aarch64_set_handled_components (sbitmap components)

>> +{

>> +  for (unsigned regno = R0_REGNUM; regno <= V31_REGNUM; regno++)

>> +    if (bitmap_bit_p (components, regno))

>> +      cfun->machine->reg_is_wrapped_separately[regno] = true;

>> +}

>> +

>>   /* AArch64 stack frames generated by this compiler look like:

>>   

>>   	+-------------------------------+

>> @@ -3944,29 +4218,6 @@ aarch64_classify_index (struct aarch64_address_info *info, rtx x,

>>     return false;

>>   }

>>   

>> -bool

>> -aarch64_offset_7bit_signed_scaled_p (machine_mode mode, HOST_WIDE_INT offset)

>> -{

>> -  return (offset >= -64 * GET_MODE_SIZE (mode)

>> -	  && offset < 64 * GET_MODE_SIZE (mode)

>> -	  && offset % GET_MODE_SIZE (mode) == 0);

>> -}

>> -

>> -static inline bool

>> -offset_9bit_signed_unscaled_p (machine_mode mode ATTRIBUTE_UNUSED,

>> -			       HOST_WIDE_INT offset)

>> -{

>> -  return offset >= -256 && offset < 256;

>> -}

>> -

>> -static inline bool

>> -offset_12bit_unsigned_scaled_p (machine_mode mode, HOST_WIDE_INT offset)

>> -{

>> -  return (offset >= 0

>> -	  && offset < 4096 * GET_MODE_SIZE (mode)

>> -	  && offset % GET_MODE_SIZE (mode) == 0);

>> -}

>> -

>>   /* Return true if MODE is one of the modes for which we

>>      support LDP/STP operations.  */

>>   

>> @@ -14452,6 +14703,30 @@ aarch64_optab_supported_p (int op, machine_mode mode1, machine_mode,

>>   #define TARGET_SCHED_FIRST_CYCLE_MULTIPASS_DFA_LOOKAHEAD_GUARD \

>>     aarch64_first_cycle_multipass_dfa_lookahead_guard

>>   

>> +#undef TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS

>> +#define TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS \

>> +  aarch64_get_separate_components

>> +

>> +#undef TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB

>> +#define TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB \

>> +  aarch64_components_for_bb

>> +

>> +#undef TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS

>> +#define TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS \

>> +  aarch64_disqualify_components

>> +

>> +#undef TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS

>> +#define TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS \

>> +  aarch64_emit_prologue_components

>> +

>> +#undef TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS

>> +#define TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS \

>> +  aarch64_emit_epilogue_components

>> +

>> +#undef TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS

>> +#define TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS \

>> +  aarch64_set_handled_components

>> +

>>   #undef TARGET_TRAMPOLINE_INIT

>>   #define TARGET_TRAMPOLINE_INIT aarch64_trampoline_init

>>   

>> diff --git a/gcc/config/aarch64/aarch64.h b/gcc/config/aarch64/aarch64.h

>> index 584ff5c..fb89e5a 100644

>> --- a/gcc/config/aarch64/aarch64.h

>> +++ b/gcc/config/aarch64/aarch64.h

>> @@ -591,6 +591,8 @@ struct GTY (()) aarch64_frame

>>   typedef struct GTY (()) machine_function

>>   {

>>     struct aarch64_frame frame;

>> +  /* One entry for each GPR and FP register.  */

>> +  bool reg_is_wrapped_separately[V31_REGNUM + 1];

> Another hardcoded use of V31_REGNUM.

>

> Thanks,

> James

>
Kyrill Tkachov Nov. 29, 2016, 11:32 a.m. UTC | #13
On 29/11/16 11:18, Kyrill Tkachov wrote:
> Hi James,

>

> On 29/11/16 10:57, James Greenhalgh wrote:

>> On Mon, Nov 14, 2016 at 02:25:28PM +0000, Kyrill Tkachov wrote:

>>> On 11/11/16 15:31, Kyrill Tkachov wrote:

>>>> On 11/11/16 10:17, Kyrill Tkachov wrote:

>>>>> On 10/11/16 23:39, Segher Boessenkool wrote:

>>>>>> On Thu, Nov 10, 2016 at 02:42:24PM -0800, Andrew Pinski wrote:

>>>>>>> On Thu, Nov 10, 2016 at 6:25 AM, Kyrill Tkachov

>>>>>>>> I ran SPEC2006 on a Cortex-A72. Overall scores were neutral but there were

>>>>>>>> some interesting swings.

>>>>>>>> 458.sjeng     +1.45%

>>>>>>>> 471.omnetpp   +2.19%

>>>>>>>> 445.gobmk     -2.01%

>>>>>>>>

>>>>>>>> On SPECFP:

>>>>>>>> 453.povray    +7.00%

>>>>>>> Wow, this looks really good.  Thank you for implementing this.  If I

>>>>>>> get some time I am going to try it out on other processors than A72

>>>>>>> but I doubt I have time any time soon.

>>>>>> I'd love to hear what causes the slowdown for gobmk as well, btw.

>>>>> I haven't yet gotten a direct answer for that (through performance analysis

>>>>> tools) but I have noticed that load/store pairs are not generated as

>>>>> aggressively as I hoped.  They are being merged by the sched fusion pass

>>>>> and peepholes (which runs after this) but it still misses cases. I've

>>>>> hacked the SWS hooks to generate pairs explicitly and that increases the

>>>>> number of pairs and helps code size to boot. It complicates the logic of

>>>>> the hooks a bit but not too much.

>>>>>

>>>>> I'll make those changes and re-benchmark, hopefully that

>>>>> will help performance.

>>>>>

>>>> And here's a version that explicitly emits pairs. I've looked at assembly

>>>> codegen on SPEC2006 and it generates quite a few more LDP/STP pairs than the

>>>> original version.  I kicked off benchmarks over the weekend to see the

>>>> effect.  Andrew, if you want to try it out (more benchmarking and testing

>>>> always welcome) this is the one to try.

>>>>

>>> And I discovered over the weekend that gamess and wrf have validation errors.

>>> This version runs correctly.  SPECINT results were fine though and there is

>>> even a small overall gain due to sjeng and omnetpp. However, gobmk still has

>>> the regression.  I'll rerun SPECFP with this patch (it's really just a small

>>> bugfix over the previous version) and get on with analysing gobmk.

>> I have some comments in line, most of which are about hardcoding the

>> maximum register number, but at a high level this looks good to me.

>

> Thanks for having a look.

> I'll respin with the comments addressed and I have a couple of comments inline.

>

> Kyrill

>

>> Thanks,

>> James

>>

>>

>

> <snip>

>

>> +

>> +/* Implement TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB.  */

>> +

>> +static sbitmap

>> +aarch64_components_for_bb (basic_block bb)

>> +{

>> +  bitmap in = DF_LIVE_IN (bb);

>> +  bitmap gen = &DF_LIVE_BB_INFO (bb)->gen;

>> +  bitmap kill = &DF_LIVE_BB_INFO (bb)->kill;

>> +

>> +  sbitmap components = sbitmap_alloc (V31_REGNUM + 1);

>> +  bitmap_clear (components);

>> +

>> +  /* GPRs are used in a bb if they are in the IN, GEN, or KILL sets.  */

>> +  for (unsigned regno = R0_REGNUM; regno <= V31_REGNUM; regno++)

>> The use of R0_REGNUM and V31_REGNUM scare me a little bit, as we're hardcoding

>> where the end of the register file is (does this, for example, fall apart

>> with the SVE work that was recently posted). Something like a

>> LAST_HARDREG_NUM might work?

>>

>

> I think you mean FIRST_PSEUDO_REGISTER. AFAICS the compiler uses

> a loop from 0 to FIRST_PSEUDO_REGISTER to go through the hard registers

> in various places in the midend.

> I'll change it to use that, though if the way to save/restore such new registers becomes

> different from the current approach (i.e. perform a DI/DFmode memory op) the code in these

> hooks will have to be updated anyway to take it into account.

>


And actually trying to implement this blows up. The "hard" registers include CC_REGNUM, which we
definitely want to avoid 'saving/restoring'. We really just want to save the normal register
data registers, so is it okay if I leave it as it is?
The prologue/epilogue code already uses V31_REGNUM, so it would need to change anyway if new
registers are added in the future.

Kyrill

>>> +    if ((!call_used_regs[regno])

>>> +       && (bitmap_bit_p (in, regno)

>>> +       || bitmap_bit_p (gen, regno)

>>> +       || bitmap_bit_p (kill, regno)))

>>> +      bitmap_set_bit (components, regno);

>>> +

>>> +  return components;

>>> +}

>>> +

>>> +/* Implement TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS.

>>> +   Nothing to do for aarch64.  */

>>> +

>>> +static void

>>> +aarch64_disqualify_components (sbitmap, edge, sbitmap, bool)

>>> +{

>>> +}

>> Is there no default "do nothing" hook for this?

>>

>

> I don't see one defined anywhere and the documentation for TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS

> says that if it is defined, all other hooks in the group must be defined.

>

>>> +

>>> +/* Return the next set bit in BMP from START onwards.  Return the total number

>>> +   of bits in BMP if no set bit is found at or after START. */

>>> +

>>> +static unsigned int

>>> +aarch64_get_next_set_bit (sbitmap bmp, unsigned int start)

>>> +{

>>> +  unsigned int nbits = SBITMAP_SIZE (bmp);

>>> +  if (start == nbits)

>>> +    return start;

>>> +

>>> +  gcc_assert (start < nbits);

>>> +  for (unsigned int i = start; i < nbits; i++)

>>> +    if (bitmap_bit_p (bmp, i))

>>> +      return i;

>>> +

>>> +  return nbits;

>>> +}

>>> +

>>> +/* Implement TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS.  */

>>> +

>>> +static void

>>> +aarch64_emit_prologue_components (sbitmap components)

>>> +{

>>> +  rtx ptr_reg = gen_rtx_REG (Pmode, frame_pointer_needed

>>> +                 ? HARD_FRAME_POINTER_REGNUM

>>> +                 : STACK_POINTER_REGNUM);

>>> +

>>> +  unsigned total_bits = SBITMAP_SIZE (components);

>> Would this be clearer called last_regno ?

>>

>>> +  unsigned regno = aarch64_get_next_set_bit (components, R0_REGNUM);

>>> +  rtx_insn *insn = NULL;

>>> +

>>> +  while (regno != total_bits)

>>> +    {

>>> +      machine_mode mode = GP_REGNUM_P (regno) ? DImode : DFmode;

>> Comment about why this can be DFmode rather than some 128-bit mode might

>> be useful here.

>>

>>> +      rtx reg = gen_rtx_REG (mode, regno);

>>> +      HOST_WIDE_INT offset = cfun->machine->frame.reg_offset[regno];

>>> +      if (!frame_pointer_needed)

>>> +    offset += cfun->machine->frame.frame_size

>>> +          - cfun->machine->frame.hard_fp_offset;

>>> +      rtx addr = plus_constant (Pmode, ptr_reg, offset);

>>> +      rtx mem = gen_frame_mem (mode, addr);

>>> +

>>> +      rtx set = gen_rtx_SET (mem, reg);

>>> +      unsigned regno2 = aarch64_get_next_set_bit (components, regno + 1);

>>> +      /* No more registers to save after REGNO.

>>> +     Emit a single save and exit.  */

>>> +      if (regno2 == total_bits)

>>> +    {

>>> +      insn = emit_insn (set);

>>> +      RTX_FRAME_RELATED_P (insn) = 1;

>>> +      add_reg_note (insn, REG_CFA_OFFSET, copy_rtx (set));

>>> +      break;

>>> +    }

>>> +

>>> +      HOST_WIDE_INT offset2 = cfun->machine->frame.reg_offset[regno2];

>>> +      /* The next register is not of the same class or its offset is not

>>> +     mergeable with the current one into a pair.  */

>>> +      if (!satisfies_constraint_Ump (mem)

>>> +      || GP_REGNUM_P (regno) != GP_REGNUM_P (regno2)

>>> +      || (offset2 - cfun->machine->frame.reg_offset[regno])

>>> +        != GET_MODE_SIZE (DImode))

>>> +    {

>>> +      insn = emit_insn (set);

>>> +      RTX_FRAME_RELATED_P (insn) = 1;

>>> +      add_reg_note (insn, REG_CFA_OFFSET, copy_rtx (set));

>>> +

>>> +      regno = regno2;

>>> +      continue;

>>> +    }

>>> +

>>> +      /* REGNO2 can be stored in a pair with REGNO.  */

>>> +      rtx reg2 = gen_rtx_REG (mode, regno2);

>>> +      if (!frame_pointer_needed)

>>> +    offset2 += cfun->machine->frame.frame_size

>>> +          - cfun->machine->frame.hard_fp_offset;

>>> +      rtx addr2 = plus_constant (Pmode, ptr_reg, offset2);

>>> +      rtx mem2 = gen_frame_mem (mode, addr2);

>>> +      rtx set2 = gen_rtx_SET (mem2, reg2);

>>> +

>>> +      insn = emit_insn (aarch64_gen_store_pair (mode, mem, reg, mem2, reg2));

>>> +      RTX_FRAME_RELATED_P (insn) = 1;

>>> +      add_reg_note (insn, REG_CFA_OFFSET, set);

>>> +      add_reg_note (insn, REG_CFA_OFFSET, set2);

>>> +

>>> +      regno = aarch64_get_next_set_bit (components, regno2 + 1);

>>> +    }

>>> +

>>> +}

>>> +

>>> +/* Implement TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS.  */

>>> +

>>> +static void

>>> +aarch64_emit_epilogue_components (sbitmap components)

>> Given the similarity of the logic, is there any way you can refactor this

>> with the prologue code above?

>>

>>> +{

>>> +

>>> +  rtx ptr_reg = gen_rtx_REG (Pmode, frame_pointer_needed

>>> +                 ? HARD_FRAME_POINTER_REGNUM

>>> +                 : STACK_POINTER_REGNUM);

>>> +  unsigned total_bits = SBITMAP_SIZE (components);

>>> +  unsigned regno = aarch64_get_next_set_bit (components, R0_REGNUM);

>>> +  rtx_insn *insn = NULL;

>>> +

>>> +  while (regno != total_bits)

>>> +    {

>>> +      machine_mode mode = GP_REGNUM_P (regno) ? DImode : DFmode;

>>> +      rtx reg = gen_rtx_REG (mode, regno);

>>> +      HOST_WIDE_INT offset = cfun->machine->frame.reg_offset[regno];

>>> +      if (!frame_pointer_needed)

>>> +    offset += cfun->machine->frame.frame_size

>>> +          - cfun->machine->frame.hard_fp_offset;

>>> +      rtx addr = plus_constant (Pmode, ptr_reg, offset);

>>> +      rtx mem = gen_frame_mem (mode, addr);

>>> +

>>> +      unsigned regno2 = aarch64_get_next_set_bit (components, regno + 1);

>>> +      /* No more registers after REGNO to restore.

>>> +     Emit a single restore and exit.  */

>>> +      if (regno2 == total_bits)

>>> +    {

>>> +      insn = emit_move_insn (reg, mem);

>>> +      RTX_FRAME_RELATED_P (insn) = 1;

>>> +      add_reg_note (insn, REG_CFA_RESTORE, reg);

>>> +      break;

>>> +    }

>>> +

>>> +      HOST_WIDE_INT offset2 = cfun->machine->frame.reg_offset[regno2];

>>> +      /* The next register is not of the same class or its offset is not

>>> +     mergeable with the current one into a pair or the offset doesn't fit

>>> +     for a load pair.  Emit a single restore and continue from REGNO2.  */

>>> +      if (!satisfies_constraint_Ump (mem)

>>> +      || GP_REGNUM_P (regno) != GP_REGNUM_P (regno2)

>>> +      || (offset2 - cfun->machine->frame.reg_offset[regno])

>>> +        != GET_MODE_SIZE (DImode))

>>> +    {

>>> +      insn = emit_move_insn (reg, mem);

>>> +      RTX_FRAME_RELATED_P (insn) = 1;

>>> +      add_reg_note (insn, REG_CFA_RESTORE, reg);

>>> +

>>> +      regno = regno2;

>>> +      continue;

>>> +    }

>>> +

>>> +      /* REGNO2 can be loaded in a pair with REGNO.  */

>>> +      rtx reg2 = gen_rtx_REG (mode, regno2);

>>> +      if (!frame_pointer_needed)

>>> +    offset2 += cfun->machine->frame.frame_size

>>> +          - cfun->machine->frame.hard_fp_offset;

>>> +      rtx addr2 = plus_constant (Pmode, ptr_reg, offset2);

>>> +      rtx mem2 = gen_frame_mem (mode, addr2);

>>> +

>>> +      insn = emit_insn (aarch64_gen_load_pair (mode, reg, mem, reg2, mem2));

>>> +      RTX_FRAME_RELATED_P (insn) = 1;

>>> +      add_reg_note (insn, REG_CFA_RESTORE, reg);

>>> +      add_reg_note (insn, REG_CFA_RESTORE, reg2);

>>> +

>>> +      regno = aarch64_get_next_set_bit (components, regno2 + 1);

>>> +    }

>>> +}

>>> +

>>> +/* Implement TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS.  */

>>> +

>>> +static void

>>> +aarch64_set_handled_components (sbitmap components)

>>> +{

>>> +  for (unsigned regno = R0_REGNUM; regno <= V31_REGNUM; regno++)

>>> +    if (bitmap_bit_p (components, regno))

>>> +      cfun->machine->reg_is_wrapped_separately[regno] = true;

>>> +}

>>> +

>>>   /* AArch64 stack frames generated by this compiler look like:

>>>         +-------------------------------+

>>> @@ -3944,29 +4218,6 @@ aarch64_classify_index (struct aarch64_address_info *info, rtx x,

>>>     return false;

>>>   }

>>>   -bool

>>> -aarch64_offset_7bit_signed_scaled_p (machine_mode mode, HOST_WIDE_INT offset)

>>> -{

>>> -  return (offset >= -64 * GET_MODE_SIZE (mode)

>>> -      && offset < 64 * GET_MODE_SIZE (mode)

>>> -      && offset % GET_MODE_SIZE (mode) == 0);

>>> -}

>>> -

>>> -static inline bool

>>> -offset_9bit_signed_unscaled_p (machine_mode mode ATTRIBUTE_UNUSED,

>>> -                   HOST_WIDE_INT offset)

>>> -{

>>> -  return offset >= -256 && offset < 256;

>>> -}

>>> -

>>> -static inline bool

>>> -offset_12bit_unsigned_scaled_p (machine_mode mode, HOST_WIDE_INT offset)

>>> -{

>>> -  return (offset >= 0

>>> -      && offset < 4096 * GET_MODE_SIZE (mode)

>>> -      && offset % GET_MODE_SIZE (mode) == 0);

>>> -}

>>> -

>>>   /* Return true if MODE is one of the modes for which we

>>>      support LDP/STP operations.  */

>>>   @@ -14452,6 +14703,30 @@ aarch64_optab_supported_p (int op, machine_mode mode1, machine_mode,

>>>   #define TARGET_SCHED_FIRST_CYCLE_MULTIPASS_DFA_LOOKAHEAD_GUARD \

>>>     aarch64_first_cycle_multipass_dfa_lookahead_guard

>>>   +#undef TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS

>>> +#define TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS \

>>> +  aarch64_get_separate_components

>>> +

>>> +#undef TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB

>>> +#define TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB \

>>> +  aarch64_components_for_bb

>>> +

>>> +#undef TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS

>>> +#define TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS \

>>> +  aarch64_disqualify_components

>>> +

>>> +#undef TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS

>>> +#define TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS \

>>> +  aarch64_emit_prologue_components

>>> +

>>> +#undef TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS

>>> +#define TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS \

>>> +  aarch64_emit_epilogue_components

>>> +

>>> +#undef TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS

>>> +#define TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS \

>>> +  aarch64_set_handled_components

>>> +

>>>   #undef TARGET_TRAMPOLINE_INIT

>>>   #define TARGET_TRAMPOLINE_INIT aarch64_trampoline_init

>>>   diff --git a/gcc/config/aarch64/aarch64.h b/gcc/config/aarch64/aarch64.h

>>> index 584ff5c..fb89e5a 100644

>>> --- a/gcc/config/aarch64/aarch64.h

>>> +++ b/gcc/config/aarch64/aarch64.h

>>> @@ -591,6 +591,8 @@ struct GTY (()) aarch64_frame

>>>   typedef struct GTY (()) machine_function

>>>   {

>>>     struct aarch64_frame frame;

>>> +  /* One entry for each GPR and FP register.  */

>>> +  bool reg_is_wrapped_separately[V31_REGNUM + 1];

>> Another hardcoded use of V31_REGNUM.

>>

>> Thanks,

>> James

>>

>
James Greenhalgh Nov. 29, 2016, 11:37 a.m. UTC | #14
On Tue, Nov 29, 2016 at 11:32:41AM +0000, Kyrill Tkachov wrote:

<snip>

> >>+

> >>+/* Implement TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB.  */

> >>+

> >>+static sbitmap

> >>+aarch64_components_for_bb (basic_block bb)

> >>+{

> >>+  bitmap in = DF_LIVE_IN (bb);

> >>+  bitmap gen = &DF_LIVE_BB_INFO (bb)->gen;

> >>+  bitmap kill = &DF_LIVE_BB_INFO (bb)->kill;

> >>+

> >>+  sbitmap components = sbitmap_alloc (V31_REGNUM + 1);

> >>+  bitmap_clear (components);

> >>+

> >>+  /* GPRs are used in a bb if they are in the IN, GEN, or KILL sets.  */

> >>+  for (unsigned regno = R0_REGNUM; regno <= V31_REGNUM; regno++)

> >>The use of R0_REGNUM and V31_REGNUM scare me a little bit, as we're hardcoding

> >>where the end of the register file is (does this, for example, fall apart

> >>with the SVE work that was recently posted). Something like a

> >>LAST_HARDREG_NUM might work?

> >>

> >

> >I think you mean FIRST_PSEUDO_REGISTER. AFAICS the compiler uses

> >a loop from 0 to FIRST_PSEUDO_REGISTER to go through the hard registers

> >in various places in the midend.

> >I'll change it to use that, though if the way to save/restore such new registers becomes

> >different from the current approach (i.e. perform a DI/DFmode memory op) the code in these

> >hooks will have to be updated anyway to take it into account.

> >

> 

> And actually trying to implement this blows up. The "hard" registers include

> CC_REGNUM, which we definitely want to avoid 'saving/restoring'. We really

> just want to save the normal register data registers, so is it okay if I

> leave it as it is?  The prologue/epilogue code already uses V31_REGNUM, so it

> would need to change anyway if new registers are added in the future.


Well, you could always define a new constant in aarch64.md ?

   (LAST_SAVED_REGISTER  63)

Which would still save hardcoding V31_REGNUM everywhere.

And yes, the pro-/epilogue functions could also do with this fix
(preapproved).

James
Segher Boessenkool Nov. 29, 2016, 8:29 p.m. UTC | #15
Hi James, Kyrill,

On Tue, Nov 29, 2016 at 10:57:33AM +0000, James Greenhalgh wrote:
> > +static sbitmap

> > +aarch64_components_for_bb (basic_block bb)

> > +{

> > +  bitmap in = DF_LIVE_IN (bb);

> > +  bitmap gen = &DF_LIVE_BB_INFO (bb)->gen;

> > +  bitmap kill = &DF_LIVE_BB_INFO (bb)->kill;

> > +

> > +  sbitmap components = sbitmap_alloc (V31_REGNUM + 1);

> > +  bitmap_clear (components);

> > +

> > +  /* GPRs are used in a bb if they are in the IN, GEN, or KILL sets.  */

> > +  for (unsigned regno = R0_REGNUM; regno <= V31_REGNUM; regno++)

> 

> The use of R0_REGNUM and V31_REGNUM scare me a little bit, as we're hardcoding

> where the end of the register file is (does this, for example, fall apart

> with the SVE work that was recently posted). Something like a

> LAST_HARDREG_NUM might work?


Components and registers aren't the same thing (you can have components
for things that aren't just a register save, e.g. the frame setup, stack
alignment, save of some non-GPR via a GPR, PIC register setup, etc.)
The loop here should really only cover the non-volatile registers, and
there should be some translation from register number to component number
(it of course is convenient to have a 1-1 translation for the GPRs and
floating point registers).  For rs6000 many things in the backend already
use non-symbolic numbers for the FPRs and GPRs, so that is easier there.

> > +static void

> > +aarch64_disqualify_components (sbitmap, edge, sbitmap, bool)

> > +{

> > +}

> 

> Is there no default "do nothing" hook for this?


I can make the shrink-wrap code do nothing here if this hook isn't
defined, if you want?


Segher
diff mbox

Patch

commit 06ac3c30d8aa38781ee9019e60a5fcf727b85231
Author: Kyrylo Tkachov <kyrylo.tkachov@arm.com>
Date:   Tue Oct 11 09:25:54 2016 +0100

    [AArch64] Separate shrink wrapping hooks implementation

diff --git a/gcc/config/aarch64/aarch64.c b/gcc/config/aarch64/aarch64.c
index 325e725..2d33ef6 100644
--- a/gcc/config/aarch64/aarch64.c
+++ b/gcc/config/aarch64/aarch64.c
@@ -1138,7 +1138,7 @@  aarch64_is_extend_from_extract (machine_mode mode, rtx mult_imm,
 
 /* Emit an insn that's a simple single-set.  Both the operands must be
    known to be valid.  */
-inline static rtx
+inline static rtx_insn *
 emit_set_insn (rtx x, rtx y)
 {
   return emit_insn (gen_rtx_SET (x, y));
@@ -3135,6 +3135,9 @@  aarch64_save_callee_saves (machine_mode mode, HOST_WIDE_INT start_offset,
 	      || regno == cfun->machine->frame.wb_candidate2))
 	continue;
 
+      if (cfun->machine->reg_is_wrapped_separately[regno])
+       continue;
+
       reg = gen_rtx_REG (mode, regno);
       offset = start_offset + cfun->machine->frame.reg_offset[regno];
       mem = gen_mem_ref (mode, plus_constant (Pmode, stack_pointer_rtx,
@@ -3143,6 +3146,7 @@  aarch64_save_callee_saves (machine_mode mode, HOST_WIDE_INT start_offset,
       regno2 = aarch64_next_callee_save (regno + 1, limit);
 
       if (regno2 <= limit
+	  && !cfun->machine->reg_is_wrapped_separately[regno2]
 	  && ((cfun->machine->frame.reg_offset[regno] + UNITS_PER_WORD)
 	      == cfun->machine->frame.reg_offset[regno2]))
 
@@ -3191,6 +3195,9 @@  aarch64_restore_callee_saves (machine_mode mode,
        regno <= limit;
        regno = aarch64_next_callee_save (regno + 1, limit))
     {
+      if (cfun->machine->reg_is_wrapped_separately[regno])
+       continue;
+
       rtx reg, mem;
 
       if (skip_wb
@@ -3205,6 +3212,7 @@  aarch64_restore_callee_saves (machine_mode mode,
       regno2 = aarch64_next_callee_save (regno + 1, limit);
 
       if (regno2 <= limit
+	  && !cfun->machine->reg_is_wrapped_separately[regno2]
 	  && ((cfun->machine->frame.reg_offset[regno] + UNITS_PER_WORD)
 	      == cfun->machine->frame.reg_offset[regno2]))
 	{
@@ -3224,6 +3232,272 @@  aarch64_restore_callee_saves (machine_mode mode,
     }
 }
 
+static inline bool
+offset_9bit_signed_unscaled_p (machine_mode mode ATTRIBUTE_UNUSED,
+			       HOST_WIDE_INT offset)
+{
+  return offset >= -256 && offset < 256;
+}
+
+static inline bool
+offset_12bit_unsigned_scaled_p (machine_mode mode, HOST_WIDE_INT offset)
+{
+  return (offset >= 0
+	  && offset < 4096 * GET_MODE_SIZE (mode)
+	  && offset % GET_MODE_SIZE (mode) == 0);
+}
+
+bool
+aarch64_offset_7bit_signed_scaled_p (machine_mode mode, HOST_WIDE_INT offset)
+{
+  return (offset >= -64 * GET_MODE_SIZE (mode)
+	  && offset < 64 * GET_MODE_SIZE (mode)
+	  && offset % GET_MODE_SIZE (mode) == 0);
+}
+
+/* Implement TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS.  */
+
+static sbitmap
+aarch64_get_separate_components (void)
+{
+  aarch64_layout_frame ();
+
+  sbitmap components = sbitmap_alloc (V31_REGNUM + 1);
+  bitmap_clear (components);
+
+  /* The registers we need saved to the frame.  */
+  for (unsigned regno = R0_REGNUM; regno <= V31_REGNUM; regno++)
+    if (aarch64_register_saved_on_entry (regno))
+      {
+	HOST_WIDE_INT offset = cfun->machine->frame.reg_offset[regno];
+	if (!frame_pointer_needed)
+	  offset += cfun->machine->frame.frame_size
+		    - cfun->machine->frame.hard_fp_offset;
+	/* Check that we can access the stack slot of the register with one
+	   direct load with no adjustments needed.  */
+	if (offset_12bit_unsigned_scaled_p (DImode, offset))
+	  bitmap_set_bit (components, regno);
+      }
+
+  /* Don't mess with the hard frame pointer.  */
+  if (frame_pointer_needed)
+    bitmap_clear_bit (components, HARD_FRAME_POINTER_REGNUM);
+
+  unsigned reg1 = cfun->machine->frame.wb_candidate1;
+  unsigned reg2 = cfun->machine->frame.wb_candidate2;
+  /* If aarch64_layout_frame has chosen registers to store/restore with
+     writeback don't interfere with them to avoid having to output explicit
+     stack adjustment instructions.  */
+  if (reg2 != INVALID_REGNUM)
+    bitmap_clear_bit (components, reg2);
+  if (reg1 != INVALID_REGNUM)
+    bitmap_clear_bit (components, reg1);
+
+  bitmap_clear_bit (components, LR_REGNUM);
+  bitmap_clear_bit (components, SP_REGNUM);
+
+  return components;
+}
+
+/* Implement TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB.  */
+
+static sbitmap
+aarch64_components_for_bb (basic_block bb)
+{
+  bitmap in = DF_LIVE_IN (bb);
+  bitmap gen = &DF_LIVE_BB_INFO (bb)->gen;
+  bitmap kill = &DF_LIVE_BB_INFO (bb)->kill;
+
+  sbitmap components = sbitmap_alloc (V31_REGNUM + 1);
+  bitmap_clear (components);
+
+  /* GPRs are used in a bb if they are in the IN, GEN, or KILL sets.  */
+  for (unsigned regno = R0_REGNUM; regno <= V31_REGNUM; regno++)
+    if ((!call_used_regs[regno])
+       && (bitmap_bit_p (in, regno)
+	   || bitmap_bit_p (gen, regno)
+	   || bitmap_bit_p (kill, regno)))
+	  bitmap_set_bit (components, regno);
+
+  return components;
+}
+
+/* Implement TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS.
+   Nothing to do for aarch64.  */
+
+static void
+aarch64_disqualify_components (sbitmap, edge, sbitmap, bool)
+{
+}
+
+/* Return the next set bit in BMP from START onwards.  Return the total number
+   of bits in BMP if no set bit is found at or after START.  */
+
+static unsigned int
+aarch64_get_next_set_bit (sbitmap bmp, unsigned int start)
+{
+  unsigned int nbits = SBITMAP_SIZE (bmp);
+  if (start == nbits)
+    return start;
+
+  gcc_assert (start < nbits);
+  for (unsigned int i = start; i < nbits; i++)
+    if (bitmap_bit_p (bmp, i))
+      return i;
+
+  return nbits;
+}
+
+/* Implement TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS.  */
+
+static void
+aarch64_emit_prologue_components (sbitmap components)
+{
+  rtx ptr_reg = gen_rtx_REG (Pmode, frame_pointer_needed
+			     ? HARD_FRAME_POINTER_REGNUM
+			     : STACK_POINTER_REGNUM);
+
+  unsigned total_bits = SBITMAP_SIZE (components);
+  unsigned regno = aarch64_get_next_set_bit (components, R0_REGNUM);
+  rtx_insn *insn = NULL;
+
+  while (regno != total_bits)
+    {
+      machine_mode mode = GP_REGNUM_P (regno) ? DImode : DFmode;
+      rtx reg = gen_rtx_REG (mode, regno);
+      HOST_WIDE_INT offset = cfun->machine->frame.reg_offset[regno];
+      if (!frame_pointer_needed)
+	offset += cfun->machine->frame.frame_size
+		  - cfun->machine->frame.hard_fp_offset;
+      rtx addr = plus_constant (Pmode, ptr_reg, offset);
+      rtx mem = gen_frame_mem (mode, addr);
+
+      rtx set = gen_rtx_SET (mem, reg);
+      unsigned regno2 = aarch64_get_next_set_bit (components, regno + 1);
+      /* No more registers to save after REGNO.
+	 Emit a single save and exit.  */
+      if (regno2 == total_bits)
+	{
+	  insn = emit_insn (set);
+	  RTX_FRAME_RELATED_P (insn) = 1;
+	  add_reg_note (insn, REG_CFA_OFFSET, copy_rtx (set));
+	  break;
+	}
+
+      HOST_WIDE_INT offset2 = cfun->machine->frame.reg_offset[regno2];
+      /* The next register is not of the same class or its offset is not
+	 mergeable with the current one into a pair.  */
+      if (!satisfies_constraint_Ump (mem)
+	  || GP_REGNUM_P (regno) != GP_REGNUM_P (regno2)
+	  || (offset2 - cfun->machine->frame.reg_offset[regno])
+		!= GET_MODE_SIZE (DImode))
+	{
+	  insn = emit_insn (set);
+	  RTX_FRAME_RELATED_P (insn) = 1;
+	  add_reg_note (insn, REG_CFA_OFFSET, copy_rtx (set));
+
+	  regno = regno2;
+	  continue;
+	}
+
+      /* REGNO2 can be stored in a pair with REGNO.  */
+      rtx reg2 = gen_rtx_REG (mode, regno2);
+      if (!frame_pointer_needed)
+	offset2 += cfun->machine->frame.frame_size
+		  - cfun->machine->frame.hard_fp_offset;
+      rtx addr2 = plus_constant (Pmode, ptr_reg, offset2);
+      rtx mem2 = gen_frame_mem (mode, addr2);
+      rtx set2 = gen_rtx_SET (mem2, reg2);
+
+      insn = emit_insn (aarch64_gen_store_pair (mode, mem, reg, mem2, reg2));
+      RTX_FRAME_RELATED_P (insn) = 1;
+      add_reg_note (insn, REG_CFA_OFFSET, set);
+      add_reg_note (insn, REG_CFA_OFFSET, set2);
+
+      regno = aarch64_get_next_set_bit (components, regno2 + 1);
+    }
+
+}
+
+/* Implement TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS.  */
+
+static void
+aarch64_emit_epilogue_components (sbitmap components)
+{
+
+  rtx ptr_reg = gen_rtx_REG (Pmode, frame_pointer_needed
+			     ? HARD_FRAME_POINTER_REGNUM
+			     : STACK_POINTER_REGNUM);
+  unsigned total_bits = SBITMAP_SIZE (components);
+  unsigned regno = aarch64_get_next_set_bit (components, R0_REGNUM);
+  rtx_insn *insn = NULL;
+
+  while (regno != total_bits)
+    {
+      machine_mode mode = GP_REGNUM_P (regno) ? DImode : DFmode;
+      rtx reg = gen_rtx_REG (mode, regno);
+      HOST_WIDE_INT offset = cfun->machine->frame.reg_offset[regno];
+      if (!frame_pointer_needed)
+	offset += cfun->machine->frame.frame_size
+		  - cfun->machine->frame.hard_fp_offset;
+      rtx addr = plus_constant (Pmode, ptr_reg, offset);
+      rtx mem = gen_frame_mem (mode, addr);
+
+      unsigned regno2 = aarch64_get_next_set_bit (components, regno + 1);
+      /* No more registers after REGNO to restore.
+	 Emit a single restore and exit.  */
+      if (regno2 == total_bits)
+	{
+	  insn = emit_move_insn (reg, mem);
+	  RTX_FRAME_RELATED_P (insn) = 1;
+	  add_reg_note (insn, REG_CFA_RESTORE, reg);
+	  break;
+	}
+
+      HOST_WIDE_INT offset2 = cfun->machine->frame.reg_offset[regno2];
+      /* The next register is not of the same class or its offset is not
+	 mergeable with the current one into a pair or the offset doesn't fit
+	 for a load pair.  Emit a single restore and continue from REGNO2.  */
+      if (!satisfies_constraint_Ump (mem)
+	  || GP_REGNUM_P (regno) != GP_REGNUM_P (regno2)
+	  || (offset2 - cfun->machine->frame.reg_offset[regno])
+		!= GET_MODE_SIZE (DImode))
+	{
+	  insn = emit_move_insn (reg, mem);
+	  RTX_FRAME_RELATED_P (insn) = 1;
+	  add_reg_note (insn, REG_CFA_RESTORE, reg);
+
+	  regno = regno2;
+	  continue;
+	}
+
+      /* REGNO2 can be loaded in a pair with REGNO.  */
+      rtx reg2 = gen_rtx_REG (mode, regno2);
+      if (!frame_pointer_needed)
+	offset2 += cfun->machine->frame.frame_size
+		  - cfun->machine->frame.hard_fp_offset;
+      rtx addr2 = plus_constant (Pmode, ptr_reg, offset2);
+      rtx mem2 = gen_frame_mem (mode, addr2);
+
+      insn = emit_insn (aarch64_gen_load_pair (mode, reg, mem, reg2, mem2));
+      RTX_FRAME_RELATED_P (insn) = 1;
+      add_reg_note (insn, REG_CFA_RESTORE, reg);
+      add_reg_note (insn, REG_CFA_RESTORE, reg2);
+
+      regno = aarch64_get_next_set_bit (components, regno2 + 1);
+    }
+}
+
+/* Implement TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS.  */
+
+static void
+aarch64_set_handled_components (sbitmap components)
+{
+  for (unsigned regno = R0_REGNUM; regno <= V31_REGNUM; regno++)
+    if (bitmap_bit_p (components, regno))
+      cfun->machine->reg_is_wrapped_separately[regno] = true;
+}
+
 /* AArch64 stack frames generated by this compiler look like:
 
 	+-------------------------------+
@@ -3944,29 +4218,6 @@  aarch64_classify_index (struct aarch64_address_info *info, rtx x,
   return false;
 }
 
-bool
-aarch64_offset_7bit_signed_scaled_p (machine_mode mode, HOST_WIDE_INT offset)
-{
-  return (offset >= -64 * GET_MODE_SIZE (mode)
-	  && offset < 64 * GET_MODE_SIZE (mode)
-	  && offset % GET_MODE_SIZE (mode) == 0);
-}
-
-static inline bool
-offset_9bit_signed_unscaled_p (machine_mode mode ATTRIBUTE_UNUSED,
-			       HOST_WIDE_INT offset)
-{
-  return offset >= -256 && offset < 256;
-}
-
-static inline bool
-offset_12bit_unsigned_scaled_p (machine_mode mode, HOST_WIDE_INT offset)
-{
-  return (offset >= 0
-	  && offset < 4096 * GET_MODE_SIZE (mode)
-	  && offset % GET_MODE_SIZE (mode) == 0);
-}
-
 /* Return true if MODE is one of the modes for which we
    support LDP/STP operations.  */
 
@@ -14452,6 +14703,30 @@  aarch64_optab_supported_p (int op, machine_mode mode1, machine_mode,
 #define TARGET_SCHED_FIRST_CYCLE_MULTIPASS_DFA_LOOKAHEAD_GUARD \
   aarch64_first_cycle_multipass_dfa_lookahead_guard
 
+#undef TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS
+#define TARGET_SHRINK_WRAP_GET_SEPARATE_COMPONENTS \
+  aarch64_get_separate_components
+
+#undef TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB
+#define TARGET_SHRINK_WRAP_COMPONENTS_FOR_BB \
+  aarch64_components_for_bb
+
+#undef TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS
+#define TARGET_SHRINK_WRAP_DISQUALIFY_COMPONENTS \
+  aarch64_disqualify_components
+
+#undef TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS
+#define TARGET_SHRINK_WRAP_EMIT_PROLOGUE_COMPONENTS \
+  aarch64_emit_prologue_components
+
+#undef TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS
+#define TARGET_SHRINK_WRAP_EMIT_EPILOGUE_COMPONENTS \
+  aarch64_emit_epilogue_components
+
+#undef TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS
+#define TARGET_SHRINK_WRAP_SET_HANDLED_COMPONENTS \
+  aarch64_set_handled_components
+
 #undef TARGET_TRAMPOLINE_INIT
 #define TARGET_TRAMPOLINE_INIT aarch64_trampoline_init
 
diff --git a/gcc/config/aarch64/aarch64.h b/gcc/config/aarch64/aarch64.h
index 584ff5c..fb89e5a 100644
--- a/gcc/config/aarch64/aarch64.h
+++ b/gcc/config/aarch64/aarch64.h
@@ -591,6 +591,8 @@  struct GTY (()) aarch64_frame
 typedef struct GTY (()) machine_function
 {
   struct aarch64_frame frame;
+  /* One entry for each GPR and FP register.  */
+  bool reg_is_wrapped_separately[V31_REGNUM + 1];
 } machine_function;
 #endif