1
0
mirror of https://github.com/golang/go synced 2024-09-30 00:04:28 -06:00

cmd/compile: combine masking and zero extension on riscv64

Combine masking with a negative value and zero extension into a single
AND operation.

Change-Id: I0b2a735b696d65568839fc4504445eeac3d869a6
Reviewed-on: https://go-review.googlesource.com/c/go/+/428498
Reviewed-by: Joedian Reid <joedian@golang.org>
Reviewed-by: Cherry Mui <cherryyz@google.com>
TryBot-Result: Gopher Robot <gobot@golang.org>
Run-TryBot: Joel Sing <joel@sing.id.au>
This commit is contained in:
Joel Sing 2022-09-04 05:26:54 +10:00
parent bb2a96b79d
commit 0ca355318f
2 changed files with 58 additions and 0 deletions

View File

@ -667,6 +667,11 @@
(MOVHUreg x:(ANDI [c] y)) && c >= 0 && int64(uint16(c)) == c => x
(MOVWUreg x:(ANDI [c] y)) && c >= 0 && int64(uint32(c)) == c => x
// Combine masking and zero extension.
(MOVBUreg (ANDI [c] x)) && c < 0 => (ANDI [int64(uint8(c))] x)
(MOVHUreg (ANDI [c] x)) && c < 0 => (ANDI [int64(uint16(c))] x)
(MOVWUreg (ANDI [c] x)) && c < 0 => (AND (MOVDconst [int64(uint32(c))]) x)
// Avoid sign/zero extension for consts.
(MOVBreg (MOVDconst [c])) => (MOVDconst [int64(int8(c))])
(MOVHreg (MOVDconst [c])) => (MOVDconst [int64(int16(c))])

View File

@ -3635,6 +3635,23 @@ func rewriteValueRISCV64_OpRISCV64MOVBUreg(v *Value) bool {
v.copyOf(x)
return true
}
// match: (MOVBUreg (ANDI [c] x))
// cond: c < 0
// result: (ANDI [int64(uint8(c))] x)
for {
if v_0.Op != OpRISCV64ANDI {
break
}
c := auxIntToInt64(v_0.AuxInt)
x := v_0.Args[0]
if !(c < 0) {
break
}
v.reset(OpRISCV64ANDI)
v.AuxInt = int64ToAuxInt(int64(uint8(c)))
v.AddArg(x)
return true
}
// match: (MOVBUreg (MOVDconst [c]))
// result: (MOVDconst [int64(uint8(c))])
for {
@ -4341,6 +4358,23 @@ func rewriteValueRISCV64_OpRISCV64MOVHUreg(v *Value) bool {
v.copyOf(x)
return true
}
// match: (MOVHUreg (ANDI [c] x))
// cond: c < 0
// result: (ANDI [int64(uint16(c))] x)
for {
if v_0.Op != OpRISCV64ANDI {
break
}
c := auxIntToInt64(v_0.AuxInt)
x := v_0.Args[0]
if !(c < 0) {
break
}
v.reset(OpRISCV64ANDI)
v.AuxInt = int64ToAuxInt(int64(uint16(c)))
v.AddArg(x)
return true
}
// match: (MOVHUreg (MOVDconst [c]))
// result: (MOVDconst [int64(uint16(c))])
for {
@ -4825,6 +4859,7 @@ func rewriteValueRISCV64_OpRISCV64MOVWUload(v *Value) bool {
func rewriteValueRISCV64_OpRISCV64MOVWUreg(v *Value) bool {
v_0 := v.Args[0]
b := v.Block
typ := &b.Func.Config.Types
// match: (MOVWUreg x:(ANDI [c] y))
// cond: c >= 0 && int64(uint32(c)) == c
// result: x
@ -4840,6 +4875,24 @@ func rewriteValueRISCV64_OpRISCV64MOVWUreg(v *Value) bool {
v.copyOf(x)
return true
}
// match: (MOVWUreg (ANDI [c] x))
// cond: c < 0
// result: (AND (MOVDconst [int64(uint32(c))]) x)
for {
if v_0.Op != OpRISCV64ANDI {
break
}
c := auxIntToInt64(v_0.AuxInt)
x := v_0.Args[0]
if !(c < 0) {
break
}
v.reset(OpRISCV64AND)
v0 := b.NewValue0(v.Pos, OpRISCV64MOVDconst, typ.UInt64)
v0.AuxInt = int64ToAuxInt(int64(uint32(c)))
v.AddArg2(v0, x)
return true
}
// match: (MOVWUreg (MOVDconst [c]))
// result: (MOVDconst [int64(uint32(c))])
for {