cmd/compile: simplify intrinsification of TrailingZeros16 and TrailingZeros8

Decompose Ctz16 and Ctz8 within the SSA rules for LOONG64, MIPS, PPC64
and S390X, rather than having a custom intrinsic. Note that for PPC64 this
actually allows the existing Ctz16 and Ctz8 rules to be used.

Change-Id: I27a5e978f852b9d75396d2a80f5d7dfcb5ef7dd4
Reviewed-on: https://go-review.googlesource.com/c/go/+/651816
Reviewed-by: Paul Murphy <murp@ibm.com>
TryBot-Result: Gopher Robot <gobot@golang.org>
LUCI-TryBot-Result: Go LUCI <golang-scoped@luci-project-accounts.iam.gserviceaccount.com>
Reviewed-by: Michael Pratt <mpratt@google.com>
Run-TryBot: Joel Sing <joel@sing.id.au>
Reviewed-by: Keith Randall <khr@golang.org>
Reviewed-by: Keith Randall <khr@google.com>
This commit is contained in:
Joel Sing 2025-02-22 23:26:21 +11:00
parent 01ba8bfe86
commit 927fdb7843
11 changed files with 224 additions and 56 deletions

View File

@ -153,8 +153,10 @@
(BitRev16 <t> x) => (REVB2H (BITREV4B <t> x)) (BitRev16 <t> x) => (REVB2H (BITREV4B <t> x))
(BitRev32 ...) => (BITREVW ...) (BitRev32 ...) => (BITREVW ...)
(BitRev64 ...) => (BITREVV ...) (BitRev64 ...) => (BITREVV ...)
(Ctz(32|64)NonZero ...) => (Ctz(32|64) ...) (Ctz(64|32|16|8)NonZero ...) => (Ctz64 ...)
(Ctz(32|64) ...) => (CTZ(W|V) ...) (Ctz(32|64) ...) => (CTZ(W|V) ...)
(Ctz16 x) => (CTZV (OR <typ.UInt64> x (MOVVconst [1<<16])))
(Ctz8 x) => (CTZV (OR <typ.UInt64> x (MOVVconst [1<<8])))
(PopCount64 <t> x) => (MOVVfpgp <t> (VPCNT64 <typ.Float64> (MOVVgpfp <typ.Float64> x))) (PopCount64 <t> x) => (MOVVfpgp <t> (VPCNT64 <typ.Float64> (MOVVgpfp <typ.Float64> x)))
(PopCount32 <t> x) => (MOVWfpgp <t> (VPCNT32 <typ.Float32> (MOVWgpfp <typ.Float32> x))) (PopCount32 <t> x) => (MOVWfpgp <t> (VPCNT32 <typ.Float32> (MOVWgpfp <typ.Float32> x)))

View File

@ -126,12 +126,13 @@
(Sqrt ...) => (SQRTD ...) (Sqrt ...) => (SQRTD ...)
(Sqrt32 ...) => (SQRTF ...) (Sqrt32 ...) => (SQRTF ...)
// TODO: optimize this case? (Ctz(32|16|8)NonZero ...) => (Ctz32 ...)
(Ctz32NonZero ...) => (Ctz32 ...)
// count trailing zero // count trailing zero
// 32 - CLZ(x&-x - 1) // 32 - CLZ(x&-x - 1)
(Ctz32 <t> x) => (SUB (MOVWconst [32]) (CLZ <t> (SUBconst <t> [1] (AND <t> x (NEG <t> x))))) (Ctz32 <t> x) => (SUB (MOVWconst [32]) (CLZ <t> (SUBconst <t> [1] (AND <t> x (NEG <t> x)))))
(Ctz16 x) => (Ctz32 (Or32 <typ.UInt32> x (MOVWconst [1<<16])))
(Ctz8 x) => (Ctz32 (Or32 <typ.UInt32> x (MOVWconst [1<<8])))
// bit length // bit length
(BitLen32 <t> x) => (SUB (MOVWconst [32]) (CLZ <t> x)) (BitLen32 <t> x) => (SUB (MOVWconst [32]) (CLZ <t> x))

View File

@ -254,16 +254,17 @@
(MOVDaddr {sym} [n] p:(ADD x y)) && sym == nil && n == 0 => p (MOVDaddr {sym} [n] p:(ADD x y)) && sym == nil && n == 0 => p
(MOVDaddr {sym} [n] ptr) && sym == nil && n == 0 && (ptr.Op == OpArgIntReg || ptr.Op == OpPhi) => ptr (MOVDaddr {sym} [n] ptr) && sym == nil && n == 0 && (ptr.Op == OpArgIntReg || ptr.Op == OpPhi) => ptr
// TODO: optimize these cases? (Ctz(64|32|16|8)NonZero ...) => (Ctz64 ...)
(Ctz32NonZero ...) => (Ctz32 ...)
(Ctz64NonZero ...) => (Ctz64 ...)
(Ctz64 x) && buildcfg.GOPPC64<=8 => (POPCNTD (ANDN <typ.Int64> (ADDconst <typ.Int64> [-1] x) x)) (Ctz64 x) && buildcfg.GOPPC64 <= 8 => (POPCNTD (ANDN <typ.Int64> (ADDconst <typ.Int64> [-1] x) x))
(Ctz64 x) => (CNTTZD x) (Ctz32 x) && buildcfg.GOPPC64 <= 8 => (POPCNTW (MOVWZreg (ANDN <typ.Int> (ADDconst <typ.Int> [-1] x) x)))
(Ctz32 x) && buildcfg.GOPPC64<=8 => (POPCNTW (MOVWZreg (ANDN <typ.Int> (ADDconst <typ.Int> [-1] x) x))) (Ctz16 x) && buildcfg.GOPPC64 <= 8 => (POPCNTW (MOVHZreg (ANDN <typ.Int16> (ADDconst <typ.Int16> [-1] x) x)))
(Ctz32 x) => (CNTTZW (MOVWZreg x)) (Ctz8 x) && buildcfg.GOPPC64 <= 8 => (POPCNTB (MOVBZreg (ANDN <typ.UInt8> (ADDconst <typ.UInt8> [-1] x) x)))
(Ctz16 x) => (POPCNTW (MOVHZreg (ANDN <typ.Int16> (ADDconst <typ.Int16> [-1] x) x)))
(Ctz8 x) => (POPCNTB (MOVBZreg (ANDN <typ.UInt8> (ADDconst <typ.UInt8> [-1] x) x))) (Ctz64 x) && buildcfg.GOPPC64 >= 9 => (CNTTZD x)
(Ctz32 x) && buildcfg.GOPPC64 >= 9 => (CNTTZW (MOVWZreg x))
(Ctz16 x) && buildcfg.GOPPC64 >= 9 => (CNTTZD (OR <typ.UInt64> x (MOVDconst [1<<16])))
(Ctz8 x) && buildcfg.GOPPC64 >= 9 => (CNTTZD (OR <typ.UInt64> x (MOVDconst [1<<8])))
(BitLen64 x) => (SUBFCconst [64] (CNTLZD <typ.Int> x)) (BitLen64 x) => (SUBFCconst [64] (CNTLZD <typ.Int> x))
(BitLen32 x) => (SUBFCconst [32] (CNTLZW <typ.Int> x)) (BitLen32 x) => (SUBFCconst [32] (CNTLZW <typ.Int> x))

View File

@ -80,13 +80,13 @@
(OffPtr [off] ptr) && is32Bit(off) => (ADDconst [int32(off)] ptr) (OffPtr [off] ptr) && is32Bit(off) => (ADDconst [int32(off)] ptr)
(OffPtr [off] ptr) => (ADD (MOVDconst [off]) ptr) (OffPtr [off] ptr) => (ADD (MOVDconst [off]) ptr)
// TODO: optimize these cases? (Ctz(64|32|16|8)NonZero ...) => (Ctz64 ...)
(Ctz64NonZero ...) => (Ctz64 ...)
(Ctz32NonZero ...) => (Ctz32 ...)
// Ctz(x) = 64 - findLeftmostOne((x-1)&^x) // Ctz(x) = 64 - findLeftmostOne((x-1)&^x)
(Ctz64 <t> x) => (SUB (MOVDconst [64]) (FLOGR (AND <t> (SUBconst <t> [1] x) (NOT <t> x)))) (Ctz64 <t> x) => (SUB (MOVDconst [64]) (FLOGR (AND <t> (SUBconst <t> [1] x) (NOT <t> x))))
(Ctz32 <t> x) => (SUB (MOVDconst [64]) (FLOGR (MOVWZreg (ANDW <t> (SUBWconst <t> [1] x) (NOTW <t> x))))) (Ctz32 <t> x) => (SUB (MOVDconst [64]) (FLOGR (MOVWZreg (ANDW <t> (SUBWconst <t> [1] x) (NOTW <t> x)))))
(Ctz16 x) => (Ctz64 (Or64 <typ.UInt64> x (MOVDconst [1<<16])))
(Ctz8 x) => (Ctz64 (Or64 <typ.UInt64> x (MOVDconst [1<<8])))
(BitLen64 x) => (SUB (MOVDconst [64]) (FLOGR x)) (BitLen64 x) => (SUB (MOVDconst [64]) (FLOGR x))
(BitLen(32|16|8) x) => (BitLen64 (ZeroExt(32|16|8)to64 x)) (BitLen(32|16|8) x) => (BitLen64 (ZeroExt(32|16|8)to64 x))

View File

@ -189,11 +189,16 @@ func rewriteValueLOONG64(v *Value) bool {
case OpCopysign: case OpCopysign:
v.Op = OpLOONG64FCOPYSGD v.Op = OpLOONG64FCOPYSGD
return true return true
case OpCtz16:
return rewriteValueLOONG64_OpCtz16(v)
case OpCtz16NonZero:
v.Op = OpCtz64
return true
case OpCtz32: case OpCtz32:
v.Op = OpLOONG64CTZW v.Op = OpLOONG64CTZW
return true return true
case OpCtz32NonZero: case OpCtz32NonZero:
v.Op = OpCtz32 v.Op = OpCtz64
return true return true
case OpCtz64: case OpCtz64:
v.Op = OpLOONG64CTZV v.Op = OpLOONG64CTZV
@ -201,6 +206,11 @@ func rewriteValueLOONG64(v *Value) bool {
case OpCtz64NonZero: case OpCtz64NonZero:
v.Op = OpCtz64 v.Op = OpCtz64
return true return true
case OpCtz8:
return rewriteValueLOONG64_OpCtz8(v)
case OpCtz8NonZero:
v.Op = OpCtz64
return true
case OpCvt32Fto32: case OpCvt32Fto32:
v.Op = OpLOONG64TRUNCFW v.Op = OpLOONG64TRUNCFW
return true return true
@ -1242,6 +1252,40 @@ func rewriteValueLOONG64_OpConstNil(v *Value) bool {
return true return true
} }
} }
func rewriteValueLOONG64_OpCtz16(v *Value) bool {
v_0 := v.Args[0]
b := v.Block
typ := &b.Func.Config.Types
// match: (Ctz16 x)
// result: (CTZV (OR <typ.UInt64> x (MOVVconst [1<<16])))
for {
x := v_0
v.reset(OpLOONG64CTZV)
v0 := b.NewValue0(v.Pos, OpLOONG64OR, typ.UInt64)
v1 := b.NewValue0(v.Pos, OpLOONG64MOVVconst, typ.UInt64)
v1.AuxInt = int64ToAuxInt(1 << 16)
v0.AddArg2(x, v1)
v.AddArg(v0)
return true
}
}
func rewriteValueLOONG64_OpCtz8(v *Value) bool {
v_0 := v.Args[0]
b := v.Block
typ := &b.Func.Config.Types
// match: (Ctz8 x)
// result: (CTZV (OR <typ.UInt64> x (MOVVconst [1<<8])))
for {
x := v_0
v.reset(OpLOONG64CTZV)
v0 := b.NewValue0(v.Pos, OpLOONG64OR, typ.UInt64)
v1 := b.NewValue0(v.Pos, OpLOONG64MOVVconst, typ.UInt64)
v1.AuxInt = int64ToAuxInt(1 << 8)
v0.AddArg2(x, v1)
v.AddArg(v0)
return true
}
}
func rewriteValueLOONG64_OpDiv16(v *Value) bool { func rewriteValueLOONG64_OpDiv16(v *Value) bool {
v_1 := v.Args[1] v_1 := v.Args[1]
v_0 := v.Args[0] v_0 := v.Args[0]

View File

@ -113,11 +113,21 @@ func rewriteValueMIPS(v *Value) bool {
return rewriteValueMIPS_OpConstBool(v) return rewriteValueMIPS_OpConstBool(v)
case OpConstNil: case OpConstNil:
return rewriteValueMIPS_OpConstNil(v) return rewriteValueMIPS_OpConstNil(v)
case OpCtz16:
return rewriteValueMIPS_OpCtz16(v)
case OpCtz16NonZero:
v.Op = OpCtz32
return true
case OpCtz32: case OpCtz32:
return rewriteValueMIPS_OpCtz32(v) return rewriteValueMIPS_OpCtz32(v)
case OpCtz32NonZero: case OpCtz32NonZero:
v.Op = OpCtz32 v.Op = OpCtz32
return true return true
case OpCtz8:
return rewriteValueMIPS_OpCtz8(v)
case OpCtz8NonZero:
v.Op = OpCtz32
return true
case OpCvt32Fto32: case OpCvt32Fto32:
v.Op = OpMIPSTRUNCFW v.Op = OpMIPSTRUNCFW
return true return true
@ -929,6 +939,23 @@ func rewriteValueMIPS_OpConstNil(v *Value) bool {
return true return true
} }
} }
func rewriteValueMIPS_OpCtz16(v *Value) bool {
v_0 := v.Args[0]
b := v.Block
typ := &b.Func.Config.Types
// match: (Ctz16 x)
// result: (Ctz32 (Or32 <typ.UInt32> x (MOVWconst [1<<16])))
for {
x := v_0
v.reset(OpCtz32)
v0 := b.NewValue0(v.Pos, OpOr32, typ.UInt32)
v1 := b.NewValue0(v.Pos, OpMIPSMOVWconst, typ.UInt32)
v1.AuxInt = int32ToAuxInt(1 << 16)
v0.AddArg2(x, v1)
v.AddArg(v0)
return true
}
}
func rewriteValueMIPS_OpCtz32(v *Value) bool { func rewriteValueMIPS_OpCtz32(v *Value) bool {
v_0 := v.Args[0] v_0 := v.Args[0]
b := v.Block b := v.Block
@ -954,6 +981,23 @@ func rewriteValueMIPS_OpCtz32(v *Value) bool {
return true return true
} }
} }
func rewriteValueMIPS_OpCtz8(v *Value) bool {
v_0 := v.Args[0]
b := v.Block
typ := &b.Func.Config.Types
// match: (Ctz8 x)
// result: (Ctz32 (Or32 <typ.UInt32> x (MOVWconst [1<<8])))
for {
x := v_0
v.reset(OpCtz32)
v0 := b.NewValue0(v.Pos, OpOr32, typ.UInt32)
v1 := b.NewValue0(v.Pos, OpMIPSMOVWconst, typ.UInt32)
v1.AuxInt = int32ToAuxInt(1 << 8)
v0.AddArg2(x, v1)
v.AddArg(v0)
return true
}
}
func rewriteValueMIPS_OpDiv16(v *Value) bool { func rewriteValueMIPS_OpDiv16(v *Value) bool {
v_1 := v.Args[1] v_1 := v.Args[1]
v_0 := v.Args[0] v_0 := v.Args[0]

View File

@ -158,10 +158,13 @@ func rewriteValuePPC64(v *Value) bool {
return rewriteValuePPC64_OpCopysign(v) return rewriteValuePPC64_OpCopysign(v)
case OpCtz16: case OpCtz16:
return rewriteValuePPC64_OpCtz16(v) return rewriteValuePPC64_OpCtz16(v)
case OpCtz16NonZero:
v.Op = OpCtz64
return true
case OpCtz32: case OpCtz32:
return rewriteValuePPC64_OpCtz32(v) return rewriteValuePPC64_OpCtz32(v)
case OpCtz32NonZero: case OpCtz32NonZero:
v.Op = OpCtz32 v.Op = OpCtz64
return true return true
case OpCtz64: case OpCtz64:
return rewriteValuePPC64_OpCtz64(v) return rewriteValuePPC64_OpCtz64(v)
@ -170,6 +173,9 @@ func rewriteValuePPC64(v *Value) bool {
return true return true
case OpCtz8: case OpCtz8:
return rewriteValuePPC64_OpCtz8(v) return rewriteValuePPC64_OpCtz8(v)
case OpCtz8NonZero:
v.Op = OpCtz64
return true
case OpCvt32Fto32: case OpCvt32Fto32:
return rewriteValuePPC64_OpCvt32Fto32(v) return rewriteValuePPC64_OpCvt32Fto32(v)
case OpCvt32Fto64: case OpCvt32Fto64:
@ -1534,9 +1540,13 @@ func rewriteValuePPC64_OpCtz16(v *Value) bool {
b := v.Block b := v.Block
typ := &b.Func.Config.Types typ := &b.Func.Config.Types
// match: (Ctz16 x) // match: (Ctz16 x)
// cond: buildcfg.GOPPC64 <= 8
// result: (POPCNTW (MOVHZreg (ANDN <typ.Int16> (ADDconst <typ.Int16> [-1] x) x))) // result: (POPCNTW (MOVHZreg (ANDN <typ.Int16> (ADDconst <typ.Int16> [-1] x) x)))
for { for {
x := v_0 x := v_0
if !(buildcfg.GOPPC64 <= 8) {
break
}
v.reset(OpPPC64POPCNTW) v.reset(OpPPC64POPCNTW)
v0 := b.NewValue0(v.Pos, OpPPC64MOVHZreg, typ.Int64) v0 := b.NewValue0(v.Pos, OpPPC64MOVHZreg, typ.Int64)
v1 := b.NewValue0(v.Pos, OpPPC64ANDN, typ.Int16) v1 := b.NewValue0(v.Pos, OpPPC64ANDN, typ.Int16)
@ -1548,13 +1558,30 @@ func rewriteValuePPC64_OpCtz16(v *Value) bool {
v.AddArg(v0) v.AddArg(v0)
return true return true
} }
// match: (Ctz16 x)
// cond: buildcfg.GOPPC64 >= 9
// result: (CNTTZD (OR <typ.UInt64> x (MOVDconst [1<<16])))
for {
x := v_0
if !(buildcfg.GOPPC64 >= 9) {
break
}
v.reset(OpPPC64CNTTZD)
v0 := b.NewValue0(v.Pos, OpPPC64OR, typ.UInt64)
v1 := b.NewValue0(v.Pos, OpPPC64MOVDconst, typ.Int64)
v1.AuxInt = int64ToAuxInt(1 << 16)
v0.AddArg2(x, v1)
v.AddArg(v0)
return true
}
return false
} }
func rewriteValuePPC64_OpCtz32(v *Value) bool { func rewriteValuePPC64_OpCtz32(v *Value) bool {
v_0 := v.Args[0] v_0 := v.Args[0]
b := v.Block b := v.Block
typ := &b.Func.Config.Types typ := &b.Func.Config.Types
// match: (Ctz32 x) // match: (Ctz32 x)
// cond: buildcfg.GOPPC64<=8 // cond: buildcfg.GOPPC64 <= 8
// result: (POPCNTW (MOVWZreg (ANDN <typ.Int> (ADDconst <typ.Int> [-1] x) x))) // result: (POPCNTW (MOVWZreg (ANDN <typ.Int> (ADDconst <typ.Int> [-1] x) x)))
for { for {
x := v_0 x := v_0
@ -1573,22 +1600,27 @@ func rewriteValuePPC64_OpCtz32(v *Value) bool {
return true return true
} }
// match: (Ctz32 x) // match: (Ctz32 x)
// cond: buildcfg.GOPPC64 >= 9
// result: (CNTTZW (MOVWZreg x)) // result: (CNTTZW (MOVWZreg x))
for { for {
x := v_0 x := v_0
if !(buildcfg.GOPPC64 >= 9) {
break
}
v.reset(OpPPC64CNTTZW) v.reset(OpPPC64CNTTZW)
v0 := b.NewValue0(v.Pos, OpPPC64MOVWZreg, typ.Int64) v0 := b.NewValue0(v.Pos, OpPPC64MOVWZreg, typ.Int64)
v0.AddArg(x) v0.AddArg(x)
v.AddArg(v0) v.AddArg(v0)
return true return true
} }
return false
} }
func rewriteValuePPC64_OpCtz64(v *Value) bool { func rewriteValuePPC64_OpCtz64(v *Value) bool {
v_0 := v.Args[0] v_0 := v.Args[0]
b := v.Block b := v.Block
typ := &b.Func.Config.Types typ := &b.Func.Config.Types
// match: (Ctz64 x) // match: (Ctz64 x)
// cond: buildcfg.GOPPC64<=8 // cond: buildcfg.GOPPC64 <= 8
// result: (POPCNTD (ANDN <typ.Int64> (ADDconst <typ.Int64> [-1] x) x)) // result: (POPCNTD (ANDN <typ.Int64> (ADDconst <typ.Int64> [-1] x) x))
for { for {
x := v_0 x := v_0
@ -1605,22 +1637,31 @@ func rewriteValuePPC64_OpCtz64(v *Value) bool {
return true return true
} }
// match: (Ctz64 x) // match: (Ctz64 x)
// cond: buildcfg.GOPPC64 >= 9
// result: (CNTTZD x) // result: (CNTTZD x)
for { for {
x := v_0 x := v_0
if !(buildcfg.GOPPC64 >= 9) {
break
}
v.reset(OpPPC64CNTTZD) v.reset(OpPPC64CNTTZD)
v.AddArg(x) v.AddArg(x)
return true return true
} }
return false
} }
func rewriteValuePPC64_OpCtz8(v *Value) bool { func rewriteValuePPC64_OpCtz8(v *Value) bool {
v_0 := v.Args[0] v_0 := v.Args[0]
b := v.Block b := v.Block
typ := &b.Func.Config.Types typ := &b.Func.Config.Types
// match: (Ctz8 x) // match: (Ctz8 x)
// cond: buildcfg.GOPPC64 <= 8
// result: (POPCNTB (MOVBZreg (ANDN <typ.UInt8> (ADDconst <typ.UInt8> [-1] x) x))) // result: (POPCNTB (MOVBZreg (ANDN <typ.UInt8> (ADDconst <typ.UInt8> [-1] x) x)))
for { for {
x := v_0 x := v_0
if !(buildcfg.GOPPC64 <= 8) {
break
}
v.reset(OpPPC64POPCNTB) v.reset(OpPPC64POPCNTB)
v0 := b.NewValue0(v.Pos, OpPPC64MOVBZreg, typ.Int64) v0 := b.NewValue0(v.Pos, OpPPC64MOVBZreg, typ.Int64)
v1 := b.NewValue0(v.Pos, OpPPC64ANDN, typ.UInt8) v1 := b.NewValue0(v.Pos, OpPPC64ANDN, typ.UInt8)
@ -1632,6 +1673,23 @@ func rewriteValuePPC64_OpCtz8(v *Value) bool {
v.AddArg(v0) v.AddArg(v0)
return true return true
} }
// match: (Ctz8 x)
// cond: buildcfg.GOPPC64 >= 9
// result: (CNTTZD (OR <typ.UInt64> x (MOVDconst [1<<8])))
for {
x := v_0
if !(buildcfg.GOPPC64 >= 9) {
break
}
v.reset(OpPPC64CNTTZD)
v0 := b.NewValue0(v.Pos, OpPPC64OR, typ.UInt64)
v1 := b.NewValue0(v.Pos, OpPPC64MOVDconst, typ.Int64)
v1.AuxInt = int64ToAuxInt(1 << 8)
v0.AddArg2(x, v1)
v.AddArg(v0)
return true
}
return false
} }
func rewriteValuePPC64_OpCvt32Fto32(v *Value) bool { func rewriteValuePPC64_OpCvt32Fto32(v *Value) bool {
v_0 := v.Args[0] v_0 := v.Args[0]

View File

@ -139,16 +139,26 @@ func rewriteValueS390X(v *Value) bool {
return rewriteValueS390X_OpConstBool(v) return rewriteValueS390X_OpConstBool(v)
case OpConstNil: case OpConstNil:
return rewriteValueS390X_OpConstNil(v) return rewriteValueS390X_OpConstNil(v)
case OpCtz16:
return rewriteValueS390X_OpCtz16(v)
case OpCtz16NonZero:
v.Op = OpCtz64
return true
case OpCtz32: case OpCtz32:
return rewriteValueS390X_OpCtz32(v) return rewriteValueS390X_OpCtz32(v)
case OpCtz32NonZero: case OpCtz32NonZero:
v.Op = OpCtz32 v.Op = OpCtz64
return true return true
case OpCtz64: case OpCtz64:
return rewriteValueS390X_OpCtz64(v) return rewriteValueS390X_OpCtz64(v)
case OpCtz64NonZero: case OpCtz64NonZero:
v.Op = OpCtz64 v.Op = OpCtz64
return true return true
case OpCtz8:
return rewriteValueS390X_OpCtz8(v)
case OpCtz8NonZero:
v.Op = OpCtz64
return true
case OpCvt32Fto32: case OpCvt32Fto32:
v.Op = OpS390XCFEBRA v.Op = OpS390XCFEBRA
return true return true
@ -1449,6 +1459,23 @@ func rewriteValueS390X_OpConstNil(v *Value) bool {
return true return true
} }
} }
func rewriteValueS390X_OpCtz16(v *Value) bool {
v_0 := v.Args[0]
b := v.Block
typ := &b.Func.Config.Types
// match: (Ctz16 x)
// result: (Ctz64 (Or64 <typ.UInt64> x (MOVDconst [1<<16])))
for {
x := v_0
v.reset(OpCtz64)
v0 := b.NewValue0(v.Pos, OpOr64, typ.UInt64)
v1 := b.NewValue0(v.Pos, OpS390XMOVDconst, typ.UInt64)
v1.AuxInt = int64ToAuxInt(1 << 16)
v0.AddArg2(x, v1)
v.AddArg(v0)
return true
}
}
func rewriteValueS390X_OpCtz32(v *Value) bool { func rewriteValueS390X_OpCtz32(v *Value) bool {
v_0 := v.Args[0] v_0 := v.Args[0]
b := v.Block b := v.Block
@ -1501,6 +1528,23 @@ func rewriteValueS390X_OpCtz64(v *Value) bool {
return true return true
} }
} }
func rewriteValueS390X_OpCtz8(v *Value) bool {
v_0 := v.Args[0]
b := v.Block
typ := &b.Func.Config.Types
// match: (Ctz8 x)
// result: (Ctz64 (Or64 <typ.UInt64> x (MOVDconst [1<<8])))
for {
x := v_0
v.reset(OpCtz64)
v0 := b.NewValue0(v.Pos, OpOr64, typ.UInt64)
v1 := b.NewValue0(v.Pos, OpS390XMOVDconst, typ.UInt64)
v1.AuxInt = int64ToAuxInt(1 << 8)
v0.AddArg2(x, v1)
v.AddArg(v0)
return true
}
}
func rewriteValueS390X_OpDiv16(v *Value) bool { func rewriteValueS390X_OpDiv16(v *Value) bool {
v_1 := v.Args[1] v_1 := v.Args[1]
v_0 := v.Args[0] v_0 := v.Args[0]

View File

@ -899,48 +899,16 @@ func initIntrinsics(cfg *intrinsicBuildConfig) {
return s.newValue1(ssa.OpCtz32, types.Types[types.TINT], args[0]) return s.newValue1(ssa.OpCtz32, types.Types[types.TINT], args[0])
}, },
sys.AMD64, sys.I386, sys.ARM64, sys.ARM, sys.Loong64, sys.S390X, sys.MIPS, sys.PPC64, sys.Wasm) sys.AMD64, sys.I386, sys.ARM64, sys.ARM, sys.Loong64, sys.S390X, sys.MIPS, sys.PPC64, sys.Wasm)
addF("math/bits", "TrailingZeros16",
func(s *state, n *ir.CallExpr, args []*ssa.Value) *ssa.Value {
x := s.newValue1(ssa.OpZeroExt16to32, types.Types[types.TUINT32], args[0])
c := s.constInt32(types.Types[types.TUINT32], 1<<16)
y := s.newValue2(ssa.OpOr32, types.Types[types.TUINT32], x, c)
return s.newValue1(ssa.OpCtz32, types.Types[types.TINT], y)
},
sys.MIPS)
addF("math/bits", "TrailingZeros16", addF("math/bits", "TrailingZeros16",
func(s *state, n *ir.CallExpr, args []*ssa.Value) *ssa.Value { func(s *state, n *ir.CallExpr, args []*ssa.Value) *ssa.Value {
return s.newValue1(ssa.OpCtz16, types.Types[types.TINT], args[0]) return s.newValue1(ssa.OpCtz16, types.Types[types.TINT], args[0])
}, },
sys.AMD64, sys.I386, sys.ARM, sys.ARM64, sys.Wasm) sys.AMD64, sys.ARM, sys.ARM64, sys.I386, sys.MIPS, sys.Loong64, sys.PPC64, sys.S390X, sys.Wasm)
addF("math/bits", "TrailingZeros16",
func(s *state, n *ir.CallExpr, args []*ssa.Value) *ssa.Value {
x := s.newValue1(ssa.OpZeroExt16to64, types.Types[types.TUINT64], args[0])
c := s.constInt64(types.Types[types.TUINT64], 1<<16)
y := s.newValue2(ssa.OpOr64, types.Types[types.TUINT64], x, c)
return s.newValue1(ssa.OpCtz64, types.Types[types.TINT], y)
},
sys.Loong64, sys.S390X, sys.PPC64)
addF("math/bits", "TrailingZeros8",
func(s *state, n *ir.CallExpr, args []*ssa.Value) *ssa.Value {
x := s.newValue1(ssa.OpZeroExt8to32, types.Types[types.TUINT32], args[0])
c := s.constInt32(types.Types[types.TUINT32], 1<<8)
y := s.newValue2(ssa.OpOr32, types.Types[types.TUINT32], x, c)
return s.newValue1(ssa.OpCtz32, types.Types[types.TINT], y)
},
sys.MIPS)
addF("math/bits", "TrailingZeros8", addF("math/bits", "TrailingZeros8",
func(s *state, n *ir.CallExpr, args []*ssa.Value) *ssa.Value { func(s *state, n *ir.CallExpr, args []*ssa.Value) *ssa.Value {
return s.newValue1(ssa.OpCtz8, types.Types[types.TINT], args[0]) return s.newValue1(ssa.OpCtz8, types.Types[types.TINT], args[0])
}, },
sys.AMD64, sys.I386, sys.ARM, sys.ARM64, sys.Wasm) sys.AMD64, sys.ARM, sys.ARM64, sys.I386, sys.MIPS, sys.Loong64, sys.PPC64, sys.S390X, sys.Wasm)
addF("math/bits", "TrailingZeros8",
func(s *state, n *ir.CallExpr, args []*ssa.Value) *ssa.Value {
x := s.newValue1(ssa.OpZeroExt8to64, types.Types[types.TUINT64], args[0])
c := s.constInt64(types.Types[types.TUINT64], 1<<8)
y := s.newValue2(ssa.OpOr64, types.Types[types.TUINT64], x, c)
return s.newValue1(ssa.OpCtz64, types.Types[types.TINT], y)
},
sys.Loong64, sys.S390X)
alias("math/bits", "ReverseBytes64", "internal/runtime/sys", "Bswap64", all...) alias("math/bits", "ReverseBytes64", "internal/runtime/sys", "Bswap64", all...)
alias("math/bits", "ReverseBytes32", "internal/runtime/sys", "Bswap32", all...) alias("math/bits", "ReverseBytes32", "internal/runtime/sys", "Bswap32", all...)
addF("math/bits", "ReverseBytes16", addF("math/bits", "ReverseBytes16",

View File

@ -867,6 +867,7 @@ var wantIntrinsics = map[testIntrinsicKey]struct{}{
{"ppc64", "internal/runtime/sys", "OnesCount64"}: struct{}{}, {"ppc64", "internal/runtime/sys", "OnesCount64"}: struct{}{},
{"ppc64", "internal/runtime/sys", "Prefetch"}: struct{}{}, {"ppc64", "internal/runtime/sys", "Prefetch"}: struct{}{},
{"ppc64", "internal/runtime/sys", "PrefetchStreamed"}: struct{}{}, {"ppc64", "internal/runtime/sys", "PrefetchStreamed"}: struct{}{},
{"ppc64", "internal/runtime/sys", "TrailingZeros8"}: struct{}{},
{"ppc64", "internal/runtime/sys", "TrailingZeros32"}: struct{}{}, {"ppc64", "internal/runtime/sys", "TrailingZeros32"}: struct{}{},
{"ppc64", "internal/runtime/sys", "TrailingZeros64"}: struct{}{}, {"ppc64", "internal/runtime/sys", "TrailingZeros64"}: struct{}{},
{"ppc64", "math", "Abs"}: struct{}{}, {"ppc64", "math", "Abs"}: struct{}{},
@ -899,6 +900,7 @@ var wantIntrinsics = map[testIntrinsicKey]struct{}{
{"ppc64", "math/bits", "RotateLeft64"}: struct{}{}, {"ppc64", "math/bits", "RotateLeft64"}: struct{}{},
{"ppc64", "math/bits", "Sub"}: struct{}{}, {"ppc64", "math/bits", "Sub"}: struct{}{},
{"ppc64", "math/bits", "Sub64"}: struct{}{}, {"ppc64", "math/bits", "Sub64"}: struct{}{},
{"ppc64", "math/bits", "TrailingZeros8"}: struct{}{},
{"ppc64", "math/bits", "TrailingZeros16"}: struct{}{}, {"ppc64", "math/bits", "TrailingZeros16"}: struct{}{},
{"ppc64", "math/bits", "TrailingZeros32"}: struct{}{}, {"ppc64", "math/bits", "TrailingZeros32"}: struct{}{},
{"ppc64", "math/bits", "TrailingZeros64"}: struct{}{}, {"ppc64", "math/bits", "TrailingZeros64"}: struct{}{},
@ -988,6 +990,7 @@ var wantIntrinsics = map[testIntrinsicKey]struct{}{
{"ppc64le", "internal/runtime/sys", "OnesCount64"}: struct{}{}, {"ppc64le", "internal/runtime/sys", "OnesCount64"}: struct{}{},
{"ppc64le", "internal/runtime/sys", "Prefetch"}: struct{}{}, {"ppc64le", "internal/runtime/sys", "Prefetch"}: struct{}{},
{"ppc64le", "internal/runtime/sys", "PrefetchStreamed"}: struct{}{}, {"ppc64le", "internal/runtime/sys", "PrefetchStreamed"}: struct{}{},
{"ppc64le", "internal/runtime/sys", "TrailingZeros8"}: struct{}{},
{"ppc64le", "internal/runtime/sys", "TrailingZeros32"}: struct{}{}, {"ppc64le", "internal/runtime/sys", "TrailingZeros32"}: struct{}{},
{"ppc64le", "internal/runtime/sys", "TrailingZeros64"}: struct{}{}, {"ppc64le", "internal/runtime/sys", "TrailingZeros64"}: struct{}{},
{"ppc64le", "math", "Abs"}: struct{}{}, {"ppc64le", "math", "Abs"}: struct{}{},
@ -1020,6 +1023,7 @@ var wantIntrinsics = map[testIntrinsicKey]struct{}{
{"ppc64le", "math/bits", "RotateLeft64"}: struct{}{}, {"ppc64le", "math/bits", "RotateLeft64"}: struct{}{},
{"ppc64le", "math/bits", "Sub"}: struct{}{}, {"ppc64le", "math/bits", "Sub"}: struct{}{},
{"ppc64le", "math/bits", "Sub64"}: struct{}{}, {"ppc64le", "math/bits", "Sub64"}: struct{}{},
{"ppc64le", "math/bits", "TrailingZeros8"}: struct{}{},
{"ppc64le", "math/bits", "TrailingZeros16"}: struct{}{}, {"ppc64le", "math/bits", "TrailingZeros16"}: struct{}{},
{"ppc64le", "math/bits", "TrailingZeros32"}: struct{}{}, {"ppc64le", "math/bits", "TrailingZeros32"}: struct{}{},
{"ppc64le", "math/bits", "TrailingZeros64"}: struct{}{}, {"ppc64le", "math/bits", "TrailingZeros64"}: struct{}{},
@ -1340,7 +1344,7 @@ func TestIntrinsics(t *testing.T) {
for ik, _ := range wantIntrinsics { for ik, _ := range wantIntrinsics {
if _, found := gotIntrinsics[ik]; !found { if _, found := gotIntrinsics[ik]; !found {
t.Errorf("Want intrinsic %v %v.%v", ik.archName, ik.pkg, ik.fn) t.Errorf("Want missing intrinsic %v %v.%v", ik.archName, ik.pkg, ik.fn)
} }
} }
} }

View File

@ -409,7 +409,7 @@ func TrailingZeros16(n uint16) int {
// arm64:"ORR\t\\$65536","RBITW","CLZW",-"MOVHU\tR",-"RBIT\t",-"CLZ\t" // arm64:"ORR\t\\$65536","RBITW","CLZW",-"MOVHU\tR",-"RBIT\t",-"CLZ\t"
// loong64:"CTZV" // loong64:"CTZV"
// s390x:"FLOGR","OR\t\\$65536" // s390x:"FLOGR","OR\t\\$65536"
// ppc64x/power8:"POPCNTD","ORIS\\t\\$1" // ppc64x/power8:"POPCNTW","ADD\t\\$-1"
// ppc64x/power9:"CNTTZD","ORIS\\t\\$1" // ppc64x/power9:"CNTTZD","ORIS\\t\\$1"
// wasm:"I64Ctz" // wasm:"I64Ctz"
return bits.TrailingZeros16(n) return bits.TrailingZeros16(n)
@ -421,6 +421,8 @@ func TrailingZeros8(n uint8) int {
// arm:"ORR\t\\$256","CLZ",-"MOVBU\tR" // arm:"ORR\t\\$256","CLZ",-"MOVBU\tR"
// arm64:"ORR\t\\$256","RBITW","CLZW",-"MOVBU\tR",-"RBIT\t",-"CLZ\t" // arm64:"ORR\t\\$256","RBITW","CLZW",-"MOVBU\tR",-"RBIT\t",-"CLZ\t"
// loong64:"CTZV" // loong64:"CTZV"
// ppc64x/power8:"POPCNTB","ADD\t\\$-1"
// ppc64x/power9:"CNTTZD","OR\t\\$256"
// s390x:"FLOGR","OR\t\\$256" // s390x:"FLOGR","OR\t\\$256"
// wasm:"I64Ctz" // wasm:"I64Ctz"
return bits.TrailingZeros8(n) return bits.TrailingZeros8(n)