[llvm] bf987e1 - [AArch64][SME]: Custom lower select and fp_extend for streaming SVE
Hassnaa Hamdi via llvm-commits
llvm-commits at lists.llvm.org
Tue Feb 14 08:03:42 PST 2023
Author: Hassnaa Hamdi
Date: 2023-02-14T16:03:34Z
New Revision: bf987e108fcc30bc30a0ad33934ebca8f1b0cd57
URL: https://github.com/llvm/llvm-project/commit/bf987e108fcc30bc30a0ad33934ebca8f1b0cd57
DIFF: https://github.com/llvm/llvm-project/commit/bf987e108fcc30bc30a0ad33934ebca8f1b0cd57.diff
LOG: [AArch64][SME]: Custom lower select and fp_extend for streaming SVE
Custom lower select/fp_extend and update related testing files
Reviewed By: sdesmalen
Differential Revision: https://reviews.llvm.org/D143434
Added:
Modified:
llvm/lib/Target/AArch64/AArch64ISelLowering.cpp
llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-fp-extend-trunc.ll
llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-fp-select.ll
llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-int-select.ll
llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-int-vselect.ll
Removed:
################################################################################
diff --git a/llvm/lib/Target/AArch64/AArch64ISelLowering.cpp b/llvm/lib/Target/AArch64/AArch64ISelLowering.cpp
index d64aedb8ea83..ba7b54aa6e48 100644
--- a/llvm/lib/Target/AArch64/AArch64ISelLowering.cpp
+++ b/llvm/lib/Target/AArch64/AArch64ISelLowering.cpp
@@ -1779,6 +1779,7 @@ void AArch64TargetLowering::addTypeForStreamingSVE(MVT VT) {
setOperationAction(ISD::FMUL, VT, Custom);
setOperationAction(ISD::FNEARBYINT, VT, Custom);
setOperationAction(ISD::FNEG, VT, Custom);
+ setOperationAction(ISD::FP_EXTEND, VT, Custom);
setOperationAction(ISD::FP_ROUND, VT, Custom);
setOperationAction(ISD::FP_TO_SINT, VT, Custom);
setOperationAction(ISD::FP_TO_UINT, VT, Custom);
@@ -1796,6 +1797,7 @@ void AArch64TargetLowering::addTypeForStreamingSVE(MVT VT) {
setOperationAction(ISD::MULHU, VT, Custom);
setOperationAction(ISD::OR, VT, Custom);
setOperationAction(ISD::SDIV, VT, Custom);
+ setOperationAction(ISD::SELECT, VT, Custom);
setOperationAction(ISD::SETCC, VT, Custom);
setOperationAction(ISD::SHL, VT, Custom);
setOperationAction(ISD::SIGN_EXTEND, VT, Custom);
@@ -3881,7 +3883,8 @@ SDValue AArch64TargetLowering::LowerFP_EXTEND(SDValue Op,
if (VT.isScalableVector())
return LowerToPredicatedOp(Op, DAG, AArch64ISD::FP_EXTEND_MERGE_PASSTHRU);
- if (useSVEForFixedLengthVectorVT(VT))
+ if (useSVEForFixedLengthVectorVT(VT,
+ Subtarget->forceStreamingCompatibleSVE()))
return LowerFixedLengthFPExtendToSVE(Op, DAG);
assert(Op.getValueType() == MVT::f128 && "Unexpected lowering");
@@ -9282,7 +9285,8 @@ SDValue AArch64TargetLowering::LowerSELECT(SDValue Op,
return DAG.getNode(ISD::VSELECT, DL, Ty, SplatPred, TVal, FVal);
}
- if (useSVEForFixedLengthVectorVT(Ty)) {
+ if (useSVEForFixedLengthVectorVT(Ty,
+ Subtarget->forceStreamingCompatibleSVE())) {
// FIXME: Ideally this would be the same as above using i1 types, however
// for the moment we can't deal with fixed i1 vector types properly, so
// instead extend the predicate to a result type sized integer vector.
diff --git a/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-fp-extend-trunc.ll b/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-fp-extend-trunc.ll
index 926cb24e5842..a1d17ba3b94b 100644
--- a/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-fp-extend-trunc.ll
+++ b/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-fp-extend-trunc.ll
@@ -3,6 +3,81 @@
target triple = "aarch64-unknown-linux-gnu"
+;
+; FCVT H -> S; Without load instr
+;
+
+define void @fcvt_v2f16_to_v2f32(<2 x half> %a, ptr %b) #0 {
+; CHECK-LABEL: fcvt_v2f16_to_v2f32:
+; CHECK: // %bb.0:
+; CHECK-NEXT: // kill: def $d0 killed $d0 def $z0
+; CHECK-NEXT: ptrue p0.s, vl4
+; CHECK-NEXT: uunpklo z0.s, z0.h
+; CHECK-NEXT: fcvt z0.s, p0/m, z0.h
+; CHECK-NEXT: str d0, [x0]
+; CHECK-NEXT: ret
+ %res = fpext <2 x half> %a to <2 x float>
+ store <2 x float> %res, ptr %b
+ ret void
+}
+
+define void @fcvt_v4f16_to_v4f32(<4 x half> %a, ptr %b) #0 {
+; CHECK-LABEL: fcvt_v4f16_to_v4f32:
+; CHECK: // %bb.0:
+; CHECK-NEXT: // kill: def $d0 killed $d0 def $z0
+; CHECK-NEXT: ptrue p0.s, vl4
+; CHECK-NEXT: uunpklo z0.s, z0.h
+; CHECK-NEXT: fcvt z0.s, p0/m, z0.h
+; CHECK-NEXT: str q0, [x0]
+; CHECK-NEXT: ret
+ %res = fpext <4 x half> %a to <4 x float>
+ store <4 x float> %res, ptr %b
+ ret void
+}
+
+define void @fcvt_v8f16_to_v8f32(<8 x half> %a, ptr %b) #0 {
+; CHECK-LABEL: fcvt_v8f16_to_v8f32:
+; CHECK: // %bb.0:
+; CHECK-NEXT: // kill: def $q0 killed $q0 def $z0
+; CHECK-NEXT: uunpklo z1.s, z0.h
+; CHECK-NEXT: ptrue p0.s, vl4
+; CHECK-NEXT: ext z0.b, z0.b, z0.b, #8
+; CHECK-NEXT: fcvt z1.s, p0/m, z1.h
+; CHECK-NEXT: uunpklo z0.s, z0.h
+; CHECK-NEXT: fcvt z0.s, p0/m, z0.h
+; CHECK-NEXT: stp q1, q0, [x0]
+; CHECK-NEXT: ret
+ %res = fpext <8 x half> %a to <8 x float>
+ store <8 x float> %res, ptr %b
+ ret void
+}
+
+define void @fcvt_v16f16_to_v16f32(<16 x half> %a, ptr %b) #0 {
+; CHECK-LABEL: fcvt_v16f16_to_v16f32:
+; CHECK: // %bb.0:
+; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
+; CHECK-NEXT: // kill: def $q0 killed $q0 def $z0
+; CHECK-NEXT: uunpklo z2.s, z1.h
+; CHECK-NEXT: uunpklo z3.s, z0.h
+; CHECK-NEXT: ext z1.b, z1.b, z1.b, #8
+; CHECK-NEXT: ext z0.b, z0.b, z0.b, #8
+; CHECK-NEXT: ptrue p0.s, vl4
+; CHECK-NEXT: uunpklo z1.s, z1.h
+; CHECK-NEXT: uunpklo z0.s, z0.h
+; CHECK-NEXT: fcvt z1.s, p0/m, z1.h
+; CHECK-NEXT: fcvt z2.s, p0/m, z2.h
+; CHECK-NEXT: fcvt z0.s, p0/m, z0.h
+; CHECK-NEXT: stp q2, q1, [x0, #32]
+; CHECK-NEXT: movprfx z1, z3
+; CHECK-NEXT: fcvt z1.s, p0/m, z3.h
+; CHECK-NEXT: stp q1, q0, [x0]
+; CHECK-NEXT: ret
+ %res = fpext <16 x half> %a to <16 x float>
+ store <16 x float> %res, ptr %b
+ ret void
+}
+
+;
;
; FCVT H -> S
;
diff --git a/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-fp-select.ll b/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-fp-select.ll
index 06e2da875a98..8b7112913d0d 100644
--- a/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-fp-select.ll
+++ b/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-fp-select.ll
@@ -6,14 +6,13 @@ target triple = "aarch64-unknown-linux-gnu"
define <2 x half> @select_v2f16(<2 x half> %op1, <2 x half> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v2f16:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
+; CHECK-NEXT: and w8, w0, #0x1
+; CHECK-NEXT: ptrue p0.h
; CHECK-NEXT: // kill: def $d0 killed $d0 def $z0
-; CHECK-NEXT: csetm w8, ne
+; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
; CHECK-NEXT: mov z2.h, w8
-; CHECK-NEXT: bic z1.d, z1.d, z2.d
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.h, p0/z, z2.h, #0
+; CHECK-NEXT: sel z0.h, p0, z0.h, z1.h
; CHECK-NEXT: // kill: def $d0 killed $d0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <2 x half> %op1, <2 x half> %op2
@@ -23,14 +22,13 @@ define <2 x half> @select_v2f16(<2 x half> %op1, <2 x half> %op2, i1 %mask) #0 {
define <4 x half> @select_v4f16(<4 x half> %op1, <4 x half> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v4f16:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
+; CHECK-NEXT: and w8, w0, #0x1
+; CHECK-NEXT: ptrue p0.h
; CHECK-NEXT: // kill: def $d0 killed $d0 def $z0
-; CHECK-NEXT: csetm w8, ne
+; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
; CHECK-NEXT: mov z2.h, w8
-; CHECK-NEXT: bic z1.d, z1.d, z2.d
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.h, p0/z, z2.h, #0
+; CHECK-NEXT: sel z0.h, p0, z0.h, z1.h
; CHECK-NEXT: // kill: def $d0 killed $d0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <4 x half> %op1, <4 x half> %op2
@@ -40,14 +38,13 @@ define <4 x half> @select_v4f16(<4 x half> %op1, <4 x half> %op2, i1 %mask) #0 {
define <8 x half> @select_v8f16(<8 x half> %op1, <8 x half> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v8f16:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
+; CHECK-NEXT: and w8, w0, #0x1
+; CHECK-NEXT: ptrue p0.h
; CHECK-NEXT: // kill: def $q0 killed $q0 def $z0
-; CHECK-NEXT: csetm w8, ne
+; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
; CHECK-NEXT: mov z2.h, w8
-; CHECK-NEXT: bic z1.d, z1.d, z2.d
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.h, p0/z, z2.h, #0
+; CHECK-NEXT: sel z0.h, p0, z0.h, z1.h
; CHECK-NEXT: // kill: def $q0 killed $q0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <8 x half> %op1, <8 x half> %op2
@@ -57,19 +54,16 @@ define <8 x half> @select_v8f16(<8 x half> %op1, <8 x half> %op2, i1 %mask) #0 {
define void @select_v16f16(ptr %a, ptr %b, i1 %mask) #0 {
; CHECK-LABEL: select_v16f16:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w2, #0x1
+; CHECK-NEXT: and w8, w2, #0x1
; CHECK-NEXT: ldr q0, [x0]
-; CHECK-NEXT: csetm w8, ne
; CHECK-NEXT: ldr q1, [x0, #16]
-; CHECK-NEXT: ldr q2, [x1, #16]
-; CHECK-NEXT: ldr q3, [x1]
+; CHECK-NEXT: ptrue p0.h
+; CHECK-NEXT: ldr q2, [x1]
+; CHECK-NEXT: ldr q3, [x1, #16]
; CHECK-NEXT: mov z4.h, w8
-; CHECK-NEXT: bic z2.d, z2.d, z4.d
-; CHECK-NEXT: and z0.d, z0.d, z4.d
-; CHECK-NEXT: bic z3.d, z3.d, z4.d
-; CHECK-NEXT: and z1.d, z1.d, z4.d
-; CHECK-NEXT: orr z0.d, z0.d, z3.d
-; CHECK-NEXT: orr z1.d, z1.d, z2.d
+; CHECK-NEXT: cmpne p0.h, p0/z, z4.h, #0
+; CHECK-NEXT: sel z0.h, p0, z0.h, z2.h
+; CHECK-NEXT: sel z1.h, p0, z1.h, z3.h
; CHECK-NEXT: stp q0, q1, [x0]
; CHECK-NEXT: ret
%op1 = load volatile <16 x half>, ptr %a
@@ -82,16 +76,13 @@ define void @select_v16f16(ptr %a, ptr %b, i1 %mask) #0 {
define <2 x float> @select_v2f32(<2 x float> %op1, <2 x float> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v2f32:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
+; CHECK-NEXT: and w8, w0, #0x1
+; CHECK-NEXT: ptrue p0.s
; CHECK-NEXT: // kill: def $d0 killed $d0 def $z0
-; CHECK-NEXT: csetm w8, ne
-; CHECK-NEXT: mvn w9, w8
+; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
; CHECK-NEXT: mov z2.s, w8
-; CHECK-NEXT: mov z3.s, w9
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: and z1.d, z1.d, z3.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.s, p0/z, z2.s, #0
+; CHECK-NEXT: sel z0.s, p0, z0.s, z1.s
; CHECK-NEXT: // kill: def $d0 killed $d0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <2 x float> %op1, <2 x float> %op2
@@ -101,16 +92,13 @@ define <2 x float> @select_v2f32(<2 x float> %op1, <2 x float> %op2, i1 %mask) #
define <4 x float> @select_v4f32(<4 x float> %op1, <4 x float> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v4f32:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
+; CHECK-NEXT: and w8, w0, #0x1
+; CHECK-NEXT: ptrue p0.s
; CHECK-NEXT: // kill: def $q0 killed $q0 def $z0
-; CHECK-NEXT: csetm w8, ne
-; CHECK-NEXT: mvn w9, w8
+; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
; CHECK-NEXT: mov z2.s, w8
-; CHECK-NEXT: mov z3.s, w9
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: and z1.d, z1.d, z3.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.s, p0/z, z2.s, #0
+; CHECK-NEXT: sel z0.s, p0, z0.s, z1.s
; CHECK-NEXT: // kill: def $q0 killed $q0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <4 x float> %op1, <4 x float> %op2
@@ -120,22 +108,17 @@ define <4 x float> @select_v4f32(<4 x float> %op1, <4 x float> %op2, i1 %mask) #
define void @select_v8f32(ptr %a, ptr %b, i1 %mask) #0 {
; CHECK-LABEL: select_v8f32:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w2, #0x1
-; CHECK-NEXT: ldr q0, [x0, #16]
-; CHECK-NEXT: csetm w8, ne
-; CHECK-NEXT: ldr q1, [x0]
-; CHECK-NEXT: mvn w9, w8
-; CHECK-NEXT: ldr q2, [x1, #16]
-; CHECK-NEXT: ldr q3, [x1]
+; CHECK-NEXT: and w8, w2, #0x1
+; CHECK-NEXT: ldr q0, [x0]
+; CHECK-NEXT: ldr q1, [x0, #16]
+; CHECK-NEXT: ptrue p0.s
+; CHECK-NEXT: ldr q2, [x1]
+; CHECK-NEXT: ldr q3, [x1, #16]
; CHECK-NEXT: mov z4.s, w8
-; CHECK-NEXT: mov z5.s, w9
-; CHECK-NEXT: and z0.d, z0.d, z4.d
-; CHECK-NEXT: and z1.d, z1.d, z4.d
-; CHECK-NEXT: and z3.d, z3.d, z5.d
-; CHECK-NEXT: and z2.d, z2.d, z5.d
-; CHECK-NEXT: orr z1.d, z1.d, z3.d
-; CHECK-NEXT: orr z0.d, z0.d, z2.d
-; CHECK-NEXT: stp q1, q0, [x0]
+; CHECK-NEXT: cmpne p0.s, p0/z, z4.s, #0
+; CHECK-NEXT: sel z0.s, p0, z0.s, z2.s
+; CHECK-NEXT: sel z1.s, p0, z1.s, z3.s
+; CHECK-NEXT: stp q0, q1, [x0]
; CHECK-NEXT: ret
%op1 = load volatile <8 x float>, ptr %a
%op2 = load volatile <8 x float>, ptr %b
@@ -166,16 +149,14 @@ define <1 x double> @select_v1f64(<1 x double> %op1, <1 x double> %op2, i1 %mask
define <2 x double> @select_v2f64(<2 x double> %op1, <2 x double> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v2f64:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
+; CHECK-NEXT: // kill: def $w0 killed $w0 def $x0
+; CHECK-NEXT: and x8, x0, #0x1
+; CHECK-NEXT: ptrue p0.d
; CHECK-NEXT: // kill: def $q0 killed $q0 def $z0
-; CHECK-NEXT: csetm x8, ne
-; CHECK-NEXT: mvn x9, x8
+; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
; CHECK-NEXT: mov z2.d, x8
-; CHECK-NEXT: mov z3.d, x9
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: and z1.d, z1.d, z3.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.d, p0/z, z2.d, #0
+; CHECK-NEXT: sel z0.d, p0, z0.d, z1.d
; CHECK-NEXT: // kill: def $q0 killed $q0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <2 x double> %op1, <2 x double> %op2
@@ -185,22 +166,18 @@ define <2 x double> @select_v2f64(<2 x double> %op1, <2 x double> %op2, i1 %mask
define void @select_v4f64(ptr %a, ptr %b, i1 %mask) #0 {
; CHECK-LABEL: select_v4f64:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w2, #0x1
-; CHECK-NEXT: ldr q0, [x0, #16]
-; CHECK-NEXT: csetm x8, ne
-; CHECK-NEXT: ldr q1, [x0]
-; CHECK-NEXT: mvn x9, x8
-; CHECK-NEXT: ldr q2, [x1, #16]
-; CHECK-NEXT: ldr q3, [x1]
+; CHECK-NEXT: // kill: def $w2 killed $w2 def $x2
+; CHECK-NEXT: and x8, x2, #0x1
+; CHECK-NEXT: ldr q0, [x0]
+; CHECK-NEXT: ldr q1, [x0, #16]
+; CHECK-NEXT: ptrue p0.d
+; CHECK-NEXT: ldr q2, [x1]
+; CHECK-NEXT: ldr q3, [x1, #16]
; CHECK-NEXT: mov z4.d, x8
-; CHECK-NEXT: mov z5.d, x9
-; CHECK-NEXT: and z0.d, z0.d, z4.d
-; CHECK-NEXT: and z1.d, z1.d, z4.d
-; CHECK-NEXT: and z3.d, z3.d, z5.d
-; CHECK-NEXT: and z2.d, z2.d, z5.d
-; CHECK-NEXT: orr z1.d, z1.d, z3.d
-; CHECK-NEXT: orr z0.d, z0.d, z2.d
-; CHECK-NEXT: stp q1, q0, [x0]
+; CHECK-NEXT: cmpne p0.d, p0/z, z4.d, #0
+; CHECK-NEXT: sel z0.d, p0, z0.d, z2.d
+; CHECK-NEXT: sel z1.d, p0, z1.d, z3.d
+; CHECK-NEXT: stp q0, q1, [x0]
; CHECK-NEXT: ret
%op1 = load volatile <4 x double>, ptr %a
%op2 = load volatile <4 x double>, ptr %b
diff --git a/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-int-select.ll b/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-int-select.ll
index e26dac067e04..b0f9a7c1f476 100644
--- a/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-int-select.ll
+++ b/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-int-select.ll
@@ -6,14 +6,13 @@ target triple = "aarch64-unknown-linux-gnu"
define <4 x i8> @select_v4i8(<4 x i8> %op1, <4 x i8> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v4i8:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
+; CHECK-NEXT: and w8, w0, #0x1
+; CHECK-NEXT: ptrue p0.h
; CHECK-NEXT: // kill: def $d0 killed $d0 def $z0
-; CHECK-NEXT: csetm w8, ne
+; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
; CHECK-NEXT: mov z2.h, w8
-; CHECK-NEXT: bic z1.d, z1.d, z2.d
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.h, p0/z, z2.h, #0
+; CHECK-NEXT: sel z0.h, p0, z0.h, z1.h
; CHECK-NEXT: // kill: def $d0 killed $d0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <4 x i8> %op1, <4 x i8> %op2
@@ -23,14 +22,13 @@ define <4 x i8> @select_v4i8(<4 x i8> %op1, <4 x i8> %op2, i1 %mask) #0 {
define <8 x i8> @select_v8i8(<8 x i8> %op1, <8 x i8> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v8i8:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
+; CHECK-NEXT: and w8, w0, #0x1
+; CHECK-NEXT: ptrue p0.b
; CHECK-NEXT: // kill: def $d0 killed $d0 def $z0
-; CHECK-NEXT: csetm w8, ne
+; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
; CHECK-NEXT: mov z2.b, w8
-; CHECK-NEXT: bic z1.d, z1.d, z2.d
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.b, p0/z, z2.b, #0
+; CHECK-NEXT: sel z0.b, p0, z0.b, z1.b
; CHECK-NEXT: // kill: def $d0 killed $d0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <8 x i8> %op1, <8 x i8> %op2
@@ -40,14 +38,13 @@ define <8 x i8> @select_v8i8(<8 x i8> %op1, <8 x i8> %op2, i1 %mask) #0 {
define <16 x i8> @select_v16i8(<16 x i8> %op1, <16 x i8> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v16i8:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
+; CHECK-NEXT: and w8, w0, #0x1
+; CHECK-NEXT: ptrue p0.b
; CHECK-NEXT: // kill: def $q0 killed $q0 def $z0
-; CHECK-NEXT: csetm w8, ne
+; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
; CHECK-NEXT: mov z2.b, w8
-; CHECK-NEXT: bic z1.d, z1.d, z2.d
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.b, p0/z, z2.b, #0
+; CHECK-NEXT: sel z0.b, p0, z0.b, z1.b
; CHECK-NEXT: // kill: def $q0 killed $q0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <16 x i8> %op1, <16 x i8> %op2
@@ -57,19 +54,16 @@ define <16 x i8> @select_v16i8(<16 x i8> %op1, <16 x i8> %op2, i1 %mask) #0 {
define void @select_v32i8(ptr %a, ptr %b, i1 %mask) #0 {
; CHECK-LABEL: select_v32i8:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w2, #0x1
+; CHECK-NEXT: and w8, w2, #0x1
; CHECK-NEXT: ldr q0, [x0]
-; CHECK-NEXT: csetm w8, ne
; CHECK-NEXT: ldr q1, [x0, #16]
+; CHECK-NEXT: ptrue p0.b
; CHECK-NEXT: ldr q2, [x1]
; CHECK-NEXT: ldr q3, [x1, #16]
; CHECK-NEXT: mov z4.b, w8
-; CHECK-NEXT: bic z2.d, z2.d, z4.d
-; CHECK-NEXT: and z0.d, z0.d, z4.d
-; CHECK-NEXT: bic z3.d, z3.d, z4.d
-; CHECK-NEXT: and z1.d, z1.d, z4.d
-; CHECK-NEXT: orr z0.d, z0.d, z2.d
-; CHECK-NEXT: orr z1.d, z1.d, z3.d
+; CHECK-NEXT: cmpne p0.b, p0/z, z4.b, #0
+; CHECK-NEXT: sel z0.b, p0, z0.b, z2.b
+; CHECK-NEXT: sel z1.b, p0, z1.b, z3.b
; CHECK-NEXT: stp q0, q1, [x0]
; CHECK-NEXT: ret
%op1 = load volatile <32 x i8>, ptr %a
@@ -82,16 +76,13 @@ define void @select_v32i8(ptr %a, ptr %b, i1 %mask) #0 {
define <2 x i16> @select_v2i16(<2 x i16> %op1, <2 x i16> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v2i16:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
+; CHECK-NEXT: and w8, w0, #0x1
+; CHECK-NEXT: ptrue p0.s
; CHECK-NEXT: // kill: def $d0 killed $d0 def $z0
-; CHECK-NEXT: csetm w8, ne
-; CHECK-NEXT: mvn w9, w8
+; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
; CHECK-NEXT: mov z2.s, w8
-; CHECK-NEXT: mov z3.s, w9
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: and z1.d, z1.d, z3.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.s, p0/z, z2.s, #0
+; CHECK-NEXT: sel z0.s, p0, z0.s, z1.s
; CHECK-NEXT: // kill: def $d0 killed $d0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <2 x i16> %op1, <2 x i16> %op2
@@ -101,14 +92,13 @@ define <2 x i16> @select_v2i16(<2 x i16> %op1, <2 x i16> %op2, i1 %mask) #0 {
define <4 x i16> @select_v4i16(<4 x i16> %op1, <4 x i16> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v4i16:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
+; CHECK-NEXT: and w8, w0, #0x1
+; CHECK-NEXT: ptrue p0.h
; CHECK-NEXT: // kill: def $d0 killed $d0 def $z0
-; CHECK-NEXT: csetm w8, ne
+; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
; CHECK-NEXT: mov z2.h, w8
-; CHECK-NEXT: bic z1.d, z1.d, z2.d
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.h, p0/z, z2.h, #0
+; CHECK-NEXT: sel z0.h, p0, z0.h, z1.h
; CHECK-NEXT: // kill: def $d0 killed $d0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <4 x i16> %op1, <4 x i16> %op2
@@ -118,14 +108,13 @@ define <4 x i16> @select_v4i16(<4 x i16> %op1, <4 x i16> %op2, i1 %mask) #0 {
define <8 x i16> @select_v8i16(<8 x i16> %op1, <8 x i16> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v8i16:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
+; CHECK-NEXT: and w8, w0, #0x1
+; CHECK-NEXT: ptrue p0.h
; CHECK-NEXT: // kill: def $q0 killed $q0 def $z0
-; CHECK-NEXT: csetm w8, ne
+; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
; CHECK-NEXT: mov z2.h, w8
-; CHECK-NEXT: bic z1.d, z1.d, z2.d
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.h, p0/z, z2.h, #0
+; CHECK-NEXT: sel z0.h, p0, z0.h, z1.h
; CHECK-NEXT: // kill: def $q0 killed $q0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <8 x i16> %op1, <8 x i16> %op2
@@ -135,19 +124,16 @@ define <8 x i16> @select_v8i16(<8 x i16> %op1, <8 x i16> %op2, i1 %mask) #0 {
define void @select_v16i16(ptr %a, ptr %b, i1 %mask) #0 {
; CHECK-LABEL: select_v16i16:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w2, #0x1
+; CHECK-NEXT: and w8, w2, #0x1
; CHECK-NEXT: ldr q0, [x0]
-; CHECK-NEXT: csetm w8, ne
; CHECK-NEXT: ldr q1, [x0, #16]
+; CHECK-NEXT: ptrue p0.h
; CHECK-NEXT: ldr q2, [x1]
; CHECK-NEXT: ldr q3, [x1, #16]
; CHECK-NEXT: mov z4.h, w8
-; CHECK-NEXT: bic z2.d, z2.d, z4.d
-; CHECK-NEXT: and z0.d, z0.d, z4.d
-; CHECK-NEXT: bic z3.d, z3.d, z4.d
-; CHECK-NEXT: and z1.d, z1.d, z4.d
-; CHECK-NEXT: orr z0.d, z0.d, z2.d
-; CHECK-NEXT: orr z1.d, z1.d, z3.d
+; CHECK-NEXT: cmpne p0.h, p0/z, z4.h, #0
+; CHECK-NEXT: sel z0.h, p0, z0.h, z2.h
+; CHECK-NEXT: sel z1.h, p0, z1.h, z3.h
; CHECK-NEXT: stp q0, q1, [x0]
; CHECK-NEXT: ret
%op1 = load volatile <16 x i16>, ptr %a
@@ -160,16 +146,13 @@ define void @select_v16i16(ptr %a, ptr %b, i1 %mask) #0 {
define <2 x i32> @select_v2i32(<2 x i32> %op1, <2 x i32> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v2i32:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
+; CHECK-NEXT: and w8, w0, #0x1
+; CHECK-NEXT: ptrue p0.s
; CHECK-NEXT: // kill: def $d0 killed $d0 def $z0
-; CHECK-NEXT: csetm w8, ne
-; CHECK-NEXT: mvn w9, w8
+; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
; CHECK-NEXT: mov z2.s, w8
-; CHECK-NEXT: mov z3.s, w9
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: and z1.d, z1.d, z3.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.s, p0/z, z2.s, #0
+; CHECK-NEXT: sel z0.s, p0, z0.s, z1.s
; CHECK-NEXT: // kill: def $d0 killed $d0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <2 x i32> %op1, <2 x i32> %op2
@@ -179,16 +162,13 @@ define <2 x i32> @select_v2i32(<2 x i32> %op1, <2 x i32> %op2, i1 %mask) #0 {
define <4 x i32> @select_v4i32(<4 x i32> %op1, <4 x i32> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v4i32:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
+; CHECK-NEXT: and w8, w0, #0x1
+; CHECK-NEXT: ptrue p0.s
; CHECK-NEXT: // kill: def $q0 killed $q0 def $z0
-; CHECK-NEXT: csetm w8, ne
-; CHECK-NEXT: mvn w9, w8
+; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
; CHECK-NEXT: mov z2.s, w8
-; CHECK-NEXT: mov z3.s, w9
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: and z1.d, z1.d, z3.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.s, p0/z, z2.s, #0
+; CHECK-NEXT: sel z0.s, p0, z0.s, z1.s
; CHECK-NEXT: // kill: def $q0 killed $q0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <4 x i32> %op1, <4 x i32> %op2
@@ -198,21 +178,16 @@ define <4 x i32> @select_v4i32(<4 x i32> %op1, <4 x i32> %op2, i1 %mask) #0 {
define void @select_v8i32(ptr %a, ptr %b, i1 %mask) #0 {
; CHECK-LABEL: select_v8i32:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w2, #0x1
+; CHECK-NEXT: and w8, w2, #0x1
; CHECK-NEXT: ldr q0, [x0]
-; CHECK-NEXT: csetm w8, ne
; CHECK-NEXT: ldr q1, [x0, #16]
-; CHECK-NEXT: mvn w9, w8
+; CHECK-NEXT: ptrue p0.s
; CHECK-NEXT: ldr q2, [x1]
; CHECK-NEXT: ldr q3, [x1, #16]
; CHECK-NEXT: mov z4.s, w8
-; CHECK-NEXT: mov z5.s, w9
-; CHECK-NEXT: and z1.d, z1.d, z4.d
-; CHECK-NEXT: and z0.d, z0.d, z4.d
-; CHECK-NEXT: and z2.d, z2.d, z5.d
-; CHECK-NEXT: and z3.d, z3.d, z5.d
-; CHECK-NEXT: orr z0.d, z0.d, z2.d
-; CHECK-NEXT: orr z1.d, z1.d, z3.d
+; CHECK-NEXT: cmpne p0.s, p0/z, z4.s, #0
+; CHECK-NEXT: sel z0.s, p0, z0.s, z2.s
+; CHECK-NEXT: sel z1.s, p0, z1.s, z3.s
; CHECK-NEXT: stp q0, q1, [x0]
; CHECK-NEXT: ret
%op1 = load volatile <8 x i32>, ptr %a
@@ -225,16 +200,14 @@ define void @select_v8i32(ptr %a, ptr %b, i1 %mask) #0 {
define <1 x i64> @select_v1i64(<1 x i64> %op1, <1 x i64> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v1i64:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
+; CHECK-NEXT: // kill: def $w0 killed $w0 def $x0
+; CHECK-NEXT: and x8, x0, #0x1
+; CHECK-NEXT: ptrue p0.d
; CHECK-NEXT: // kill: def $d0 killed $d0 def $z0
-; CHECK-NEXT: csetm x8, ne
-; CHECK-NEXT: mvn x9, x8
+; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
; CHECK-NEXT: mov z2.d, x8
-; CHECK-NEXT: mov z3.d, x9
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: and z1.d, z1.d, z3.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.d, p0/z, z2.d, #0
+; CHECK-NEXT: sel z0.d, p0, z0.d, z1.d
; CHECK-NEXT: // kill: def $d0 killed $d0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <1 x i64> %op1, <1 x i64> %op2
@@ -244,16 +217,14 @@ define <1 x i64> @select_v1i64(<1 x i64> %op1, <1 x i64> %op2, i1 %mask) #0 {
define <2 x i64> @select_v2i64(<2 x i64> %op1, <2 x i64> %op2, i1 %mask) #0 {
; CHECK-LABEL: select_v2i64:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
+; CHECK-NEXT: // kill: def $w0 killed $w0 def $x0
+; CHECK-NEXT: and x8, x0, #0x1
+; CHECK-NEXT: ptrue p0.d
; CHECK-NEXT: // kill: def $q0 killed $q0 def $z0
-; CHECK-NEXT: csetm x8, ne
-; CHECK-NEXT: mvn x9, x8
+; CHECK-NEXT: // kill: def $q1 killed $q1 def $z1
; CHECK-NEXT: mov z2.d, x8
-; CHECK-NEXT: mov z3.d, x9
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: and z1.d, z1.d, z3.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.d, p0/z, z2.d, #0
+; CHECK-NEXT: sel z0.d, p0, z0.d, z1.d
; CHECK-NEXT: // kill: def $q0 killed $q0 killed $z0
; CHECK-NEXT: ret
%sel = select i1 %mask, <2 x i64> %op1, <2 x i64> %op2
@@ -263,21 +234,17 @@ define <2 x i64> @select_v2i64(<2 x i64> %op1, <2 x i64> %op2, i1 %mask) #0 {
define void @select_v4i64(ptr %a, ptr %b, i1 %mask) #0 {
; CHECK-LABEL: select_v4i64:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w2, #0x1
+; CHECK-NEXT: // kill: def $w2 killed $w2 def $x2
+; CHECK-NEXT: and x8, x2, #0x1
; CHECK-NEXT: ldr q0, [x0]
-; CHECK-NEXT: csetm x8, ne
; CHECK-NEXT: ldr q1, [x0, #16]
-; CHECK-NEXT: mvn x9, x8
+; CHECK-NEXT: ptrue p0.d
; CHECK-NEXT: ldr q2, [x1]
; CHECK-NEXT: ldr q3, [x1, #16]
; CHECK-NEXT: mov z4.d, x8
-; CHECK-NEXT: mov z5.d, x9
-; CHECK-NEXT: and z1.d, z1.d, z4.d
-; CHECK-NEXT: and z0.d, z0.d, z4.d
-; CHECK-NEXT: and z2.d, z2.d, z5.d
-; CHECK-NEXT: and z3.d, z3.d, z5.d
-; CHECK-NEXT: orr z0.d, z0.d, z2.d
-; CHECK-NEXT: orr z1.d, z1.d, z3.d
+; CHECK-NEXT: cmpne p0.d, p0/z, z4.d, #0
+; CHECK-NEXT: sel z0.d, p0, z0.d, z2.d
+; CHECK-NEXT: sel z1.d, p0, z1.d, z3.d
; CHECK-NEXT: stp q0, q1, [x0]
; CHECK-NEXT: ret
%op1 = load volatile <4 x i64>, ptr %a
diff --git a/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-int-vselect.ll b/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-int-vselect.ll
index 7d8fe130b402..3f6a9b9d8785 100644
--- a/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-int-vselect.ll
+++ b/llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-int-vselect.ll
@@ -212,16 +212,14 @@ define void @select_v8i32(ptr %a, ptr %b) #0 {
define <1 x i64> @select_v1i64(<1 x i64> %op1, <1 x i64> %op2, <1 x i1> %mask) #0 {
; CHECK-LABEL: select_v1i64:
; CHECK: // %bb.0:
-; CHECK-NEXT: tst w0, #0x1
-; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
+; CHECK-NEXT: // kill: def $w0 killed $w0 def $x0
+; CHECK-NEXT: and x8, x0, #0x1
+; CHECK-NEXT: ptrue p0.d
; CHECK-NEXT: // kill: def $d0 killed $d0 def $z0
-; CHECK-NEXT: csetm x8, ne
-; CHECK-NEXT: mvn x9, x8
+; CHECK-NEXT: // kill: def $d1 killed $d1 def $z1
; CHECK-NEXT: mov z2.d, x8
-; CHECK-NEXT: mov z3.d, x9
-; CHECK-NEXT: and z0.d, z0.d, z2.d
-; CHECK-NEXT: and z1.d, z1.d, z3.d
-; CHECK-NEXT: orr z0.d, z0.d, z1.d
+; CHECK-NEXT: cmpne p0.d, p0/z, z2.d, #0
+; CHECK-NEXT: sel z0.d, p0, z0.d, z1.d
; CHECK-NEXT: // kill: def $d0 killed $d0 killed $z0
; CHECK-NEXT: ret
%sel = select <1 x i1> %mask, <1 x i64> %op1, <1 x i64> %op2
More information about the llvm-commits
mailing list