mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-23 11:13:28 +01:00
14330a06ca
This addes extra patterns for the VMLAS MVE instruction, which performs Qda = Qda * Qn + Rm, a similar pattern to the existing VMLA. The sinking of splat(Rm) into the loop is already performed, meaning we just need extra Pat's in tablegen. Differential Revision: https://reviews.llvm.org/D75115
388 lines
14 KiB
LLVM
388 lines
14 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
|
|
; RUN: llc -mtriple=thumbv8.1m.main-arm-none-eabi -mattr=+mve.fp %s -o - | FileCheck %s
|
|
|
|
define arm_aapcs_vfpcc <4 x i32> @vmlau32(<4 x i32> %A, <4 x i32> %B, i32 %X) nounwind {
|
|
; CHECK-LABEL: vmlau32:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: vmla.u32 q0, q1, r0
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%0 = insertelement <4 x i32> undef, i32 %X, i32 0
|
|
%1 = shufflevector <4 x i32> %0, <4 x i32> undef, <4 x i32> zeroinitializer
|
|
%2 = mul nsw <4 x i32> %B, %1
|
|
%3 = add nsw <4 x i32> %A, %2
|
|
ret <4 x i32> %3
|
|
}
|
|
|
|
define arm_aapcs_vfpcc <4 x i32> @vmlau32b(<4 x i32> %A, <4 x i32> %B, i32 %X) nounwind {
|
|
; CHECK-LABEL: vmlau32b:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: vmla.u32 q0, q1, r0
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%0 = insertelement <4 x i32> undef, i32 %X, i32 0
|
|
%1 = shufflevector <4 x i32> %0, <4 x i32> undef, <4 x i32> zeroinitializer
|
|
%2 = mul nsw <4 x i32> %1, %B
|
|
%3 = add nsw <4 x i32> %2, %A
|
|
ret <4 x i32> %3
|
|
}
|
|
|
|
define arm_aapcs_vfpcc <8 x i16> @vmlau16(<8 x i16> %A, <8 x i16> %B, i16 %X) nounwind {
|
|
; CHECK-LABEL: vmlau16:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: vmla.u16 q0, q1, r0
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%0 = insertelement <8 x i16> undef, i16 %X, i32 0
|
|
%1 = shufflevector <8 x i16> %0, <8 x i16> undef, <8 x i32> zeroinitializer
|
|
%2 = mul nsw <8 x i16> %B, %1
|
|
%3 = add nsw <8 x i16> %A, %2
|
|
ret <8 x i16> %3
|
|
}
|
|
|
|
define arm_aapcs_vfpcc <8 x i16> @vmlau16b(<8 x i16> %A, <8 x i16> %B, i16 %X) nounwind {
|
|
; CHECK-LABEL: vmlau16b:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: vmla.u16 q0, q1, r0
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%0 = insertelement <8 x i16> undef, i16 %X, i32 0
|
|
%1 = shufflevector <8 x i16> %0, <8 x i16> undef, <8 x i32> zeroinitializer
|
|
%2 = mul nsw <8 x i16> %1, %B
|
|
%3 = add nsw <8 x i16> %2, %A
|
|
ret <8 x i16> %3
|
|
}
|
|
|
|
define arm_aapcs_vfpcc <16 x i8> @vmlau8(<16 x i8> %A, <16 x i8> %B, i8 %X) nounwind {
|
|
; CHECK-LABEL: vmlau8:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: vmla.u8 q0, q1, r0
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%0 = insertelement <16 x i8> undef, i8 %X, i32 0
|
|
%1 = shufflevector <16 x i8> %0, <16 x i8> undef, <16 x i32> zeroinitializer
|
|
%2 = mul nsw <16 x i8> %B, %1
|
|
%3 = add nsw <16 x i8> %A, %2
|
|
ret <16 x i8> %3
|
|
}
|
|
|
|
define arm_aapcs_vfpcc <16 x i8> @vmlau8b(<16 x i8> %A, <16 x i8> %B, i8 %X) nounwind {
|
|
; CHECK-LABEL: vmlau8b:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: vmla.u8 q0, q1, r0
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%0 = insertelement <16 x i8> undef, i8 %X, i32 0
|
|
%1 = shufflevector <16 x i8> %0, <16 x i8> undef, <16 x i32> zeroinitializer
|
|
%2 = mul nsw <16 x i8> %1, %B
|
|
%3 = add nsw <16 x i8> %2, %A
|
|
ret <16 x i8> %3
|
|
}
|
|
|
|
define void @vmla32_in_loop(i32* %s1, i32 %x, i32* %d, i32 %n) {
|
|
; CHECK-LABEL: vmla32_in_loop:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: .LBB6_1: @ %vector.body
|
|
; CHECK-NEXT: @ =>This Inner Loop Header: Depth=1
|
|
; CHECK-NEXT: vldrw.u32 q0, [r0], #16
|
|
; CHECK-NEXT: vldrw.u32 q1, [r2]
|
|
; CHECK-NEXT: subs r3, #4
|
|
; CHECK-NEXT: vmla.u32 q1, q0, r1
|
|
; CHECK-NEXT: vstrb.8 q1, [r2], #16
|
|
; CHECK-NEXT: bne .LBB6_1
|
|
; CHECK-NEXT: @ %bb.2: @ %for.cond.cleanup
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%broadcast.splatinsert8 = insertelement <4 x i32> undef, i32 %x, i32 0
|
|
%broadcast.splat9 = shufflevector <4 x i32> %broadcast.splatinsert8, <4 x i32> undef, <4 x i32> zeroinitializer
|
|
br label %vector.body
|
|
|
|
vector.body:
|
|
%index = phi i32 [ 0, %entry ], [ %index.next, %vector.body ]
|
|
%0 = getelementptr inbounds i32, i32* %s1, i32 %index
|
|
%1 = bitcast i32* %0 to <4 x i32>*
|
|
%wide.load = load <4 x i32>, <4 x i32>* %1, align 4
|
|
%2 = mul nsw <4 x i32> %wide.load, %broadcast.splat9
|
|
%3 = getelementptr inbounds i32, i32* %d, i32 %index
|
|
%4 = bitcast i32* %3 to <4 x i32>*
|
|
%wide.load10 = load <4 x i32>, <4 x i32>* %4, align 4
|
|
%5 = add nsw <4 x i32> %wide.load10, %2
|
|
%6 = bitcast i32* %3 to <4 x i32>*
|
|
store <4 x i32> %5, <4 x i32>* %6, align 4
|
|
%index.next = add i32 %index, 4
|
|
%7 = icmp eq i32 %index.next, %n
|
|
br i1 %7, label %for.cond.cleanup, label %vector.body
|
|
|
|
for.cond.cleanup:
|
|
ret void
|
|
}
|
|
|
|
define void @vmla16_in_loop(i16* %s1, i16 %x, i16* %d, i32 %n) {
|
|
; CHECK-LABEL: vmla16_in_loop:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: .LBB7_1: @ %vector.body
|
|
; CHECK-NEXT: @ =>This Inner Loop Header: Depth=1
|
|
; CHECK-NEXT: vldrh.u16 q0, [r0], #16
|
|
; CHECK-NEXT: vldrh.u16 q1, [r2]
|
|
; CHECK-NEXT: subs r3, #8
|
|
; CHECK-NEXT: vmla.u16 q1, q0, r1
|
|
; CHECK-NEXT: vstrb.8 q1, [r2], #16
|
|
; CHECK-NEXT: bne .LBB7_1
|
|
; CHECK-NEXT: @ %bb.2: @ %for.cond.cleanup
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%broadcast.splatinsert11 = insertelement <8 x i16> undef, i16 %x, i32 0
|
|
%broadcast.splat12 = shufflevector <8 x i16> %broadcast.splatinsert11, <8 x i16> undef, <8 x i32> zeroinitializer
|
|
br label %vector.body
|
|
|
|
vector.body:
|
|
%index = phi i32 [ 0, %entry ], [ %index.next, %vector.body ]
|
|
%0 = getelementptr inbounds i16, i16* %s1, i32 %index
|
|
%1 = bitcast i16* %0 to <8 x i16>*
|
|
%wide.load = load <8 x i16>, <8 x i16>* %1, align 2
|
|
%2 = mul <8 x i16> %wide.load, %broadcast.splat12
|
|
%3 = getelementptr inbounds i16, i16* %d, i32 %index
|
|
%4 = bitcast i16* %3 to <8 x i16>*
|
|
%wide.load13 = load <8 x i16>, <8 x i16>* %4, align 2
|
|
%5 = add <8 x i16> %2, %wide.load13
|
|
%6 = bitcast i16* %3 to <8 x i16>*
|
|
store <8 x i16> %5, <8 x i16>* %6, align 2
|
|
%index.next = add i32 %index, 8
|
|
%7 = icmp eq i32 %index.next, %n
|
|
br i1 %7, label %for.cond.cleanup, label %vector.body
|
|
|
|
for.cond.cleanup:
|
|
ret void
|
|
}
|
|
|
|
define void @vmla8_in_loop(i8* %s1, i8 %x, i8* %d, i32 %n) {
|
|
; CHECK-LABEL: vmla8_in_loop:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: .LBB8_1: @ %vector.body
|
|
; CHECK-NEXT: @ =>This Inner Loop Header: Depth=1
|
|
; CHECK-NEXT: vldrh.u16 q0, [r0], #16
|
|
; CHECK-NEXT: vldrh.u16 q1, [r2]
|
|
; CHECK-NEXT: subs r3, #16
|
|
; CHECK-NEXT: vmla.u8 q1, q0, r1
|
|
; CHECK-NEXT: vstrb.8 q1, [r2], #16
|
|
; CHECK-NEXT: bne .LBB8_1
|
|
; CHECK-NEXT: @ %bb.2: @ %for.cond.cleanup
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%broadcast.splatinsert11 = insertelement <16 x i8> undef, i8 %x, i32 0
|
|
%broadcast.splat12 = shufflevector <16 x i8> %broadcast.splatinsert11, <16 x i8> undef, <16 x i32> zeroinitializer
|
|
br label %vector.body
|
|
|
|
vector.body:
|
|
%index = phi i32 [ 0, %entry ], [ %index.next, %vector.body ]
|
|
%0 = getelementptr inbounds i8, i8* %s1, i32 %index
|
|
%1 = bitcast i8* %0 to <16 x i8>*
|
|
%wide.load = load <16 x i8>, <16 x i8>* %1, align 2
|
|
%2 = mul <16 x i8> %wide.load, %broadcast.splat12
|
|
%3 = getelementptr inbounds i8, i8* %d, i32 %index
|
|
%4 = bitcast i8* %3 to <16 x i8>*
|
|
%wide.load13 = load <16 x i8>, <16 x i8>* %4, align 2
|
|
%5 = add <16 x i8> %2, %wide.load13
|
|
%6 = bitcast i8* %3 to <16 x i8>*
|
|
store <16 x i8> %5, <16 x i8>* %6, align 2
|
|
%index.next = add i32 %index, 16
|
|
%7 = icmp eq i32 %index.next, %n
|
|
br i1 %7, label %for.cond.cleanup, label %vector.body
|
|
|
|
for.cond.cleanup:
|
|
ret void
|
|
}
|
|
|
|
|
|
define arm_aapcs_vfpcc <4 x i32> @vmlasu32(<4 x i32> %A, <4 x i32> %B, i32 %X) nounwind {
|
|
; CHECK-LABEL: vmlasu32:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: vmlas.u32 q0, q1, r0
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%0 = insertelement <4 x i32> undef, i32 %X, i32 0
|
|
%1 = shufflevector <4 x i32> %0, <4 x i32> undef, <4 x i32> zeroinitializer
|
|
%2 = mul nsw <4 x i32> %A, %B
|
|
%3 = add nsw <4 x i32> %1, %2
|
|
ret <4 x i32> %3
|
|
}
|
|
|
|
define arm_aapcs_vfpcc <4 x i32> @vmlasu32b(<4 x i32> %A, <4 x i32> %B, i32 %X) nounwind {
|
|
; CHECK-LABEL: vmlasu32b:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: vmlas.u32 q0, q1, r0
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%0 = insertelement <4 x i32> undef, i32 %X, i32 0
|
|
%1 = shufflevector <4 x i32> %0, <4 x i32> undef, <4 x i32> zeroinitializer
|
|
%2 = mul nsw <4 x i32> %A, %B
|
|
%3 = add nsw <4 x i32> %2, %1
|
|
ret <4 x i32> %3
|
|
}
|
|
|
|
define arm_aapcs_vfpcc <8 x i16> @vmlasu16(<8 x i16> %A, <8 x i16> %B, i16 %X) nounwind {
|
|
; CHECK-LABEL: vmlasu16:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: vmlas.u16 q0, q1, r0
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%0 = insertelement <8 x i16> undef, i16 %X, i32 0
|
|
%1 = shufflevector <8 x i16> %0, <8 x i16> undef, <8 x i32> zeroinitializer
|
|
%2 = mul nsw <8 x i16> %A, %B
|
|
%3 = add nsw <8 x i16> %1, %2
|
|
ret <8 x i16> %3
|
|
}
|
|
|
|
define arm_aapcs_vfpcc <8 x i16> @vmlasu16b(<8 x i16> %A, <8 x i16> %B, i16 %X) nounwind {
|
|
; CHECK-LABEL: vmlasu16b:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: vmlas.u16 q0, q1, r0
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%0 = insertelement <8 x i16> undef, i16 %X, i32 0
|
|
%1 = shufflevector <8 x i16> %0, <8 x i16> undef, <8 x i32> zeroinitializer
|
|
%2 = mul nsw <8 x i16> %A, %B
|
|
%3 = add nsw <8 x i16> %2, %1
|
|
ret <8 x i16> %3
|
|
}
|
|
|
|
define arm_aapcs_vfpcc <16 x i8> @vmlasu8(<16 x i8> %A, <16 x i8> %B, i8 %X) nounwind {
|
|
; CHECK-LABEL: vmlasu8:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: vmlas.u8 q0, q1, r0
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%0 = insertelement <16 x i8> undef, i8 %X, i32 0
|
|
%1 = shufflevector <16 x i8> %0, <16 x i8> undef, <16 x i32> zeroinitializer
|
|
%2 = mul nsw <16 x i8> %A, %B
|
|
%3 = add nsw <16 x i8> %1, %2
|
|
ret <16 x i8> %3
|
|
}
|
|
|
|
define arm_aapcs_vfpcc <16 x i8> @vmlasu8b(<16 x i8> %A, <16 x i8> %B, i8 %X) nounwind {
|
|
; CHECK-LABEL: vmlasu8b:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: vmlas.u8 q0, q1, r0
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%0 = insertelement <16 x i8> undef, i8 %X, i32 0
|
|
%1 = shufflevector <16 x i8> %0, <16 x i8> undef, <16 x i32> zeroinitializer
|
|
%2 = mul nsw <16 x i8> %A, %B
|
|
%3 = add nsw <16 x i8> %2, %1
|
|
ret <16 x i8> %3
|
|
}
|
|
|
|
define void @vmlas32_in_loop(i32* %s1, i32 %x, i32* %d, i32 %n) {
|
|
; CHECK-LABEL: vmlas32_in_loop:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: .LBB15_1: @ %vector.body
|
|
; CHECK-NEXT: @ =>This Inner Loop Header: Depth=1
|
|
; CHECK-NEXT: vldrw.u32 q0, [r2]
|
|
; CHECK-NEXT: vldrw.u32 q1, [r0], #16
|
|
; CHECK-NEXT: subs r3, #4
|
|
; CHECK-NEXT: vmlas.u32 q1, q0, r1
|
|
; CHECK-NEXT: vstrb.8 q1, [r2], #16
|
|
; CHECK-NEXT: bne .LBB15_1
|
|
; CHECK-NEXT: @ %bb.2: @ %for.cond.cleanup
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%broadcast.splatinsert8 = insertelement <4 x i32> undef, i32 %x, i32 0
|
|
%broadcast.splat9 = shufflevector <4 x i32> %broadcast.splatinsert8, <4 x i32> undef, <4 x i32> zeroinitializer
|
|
br label %vector.body
|
|
|
|
vector.body:
|
|
%index = phi i32 [ 0, %entry ], [ %index.next, %vector.body ]
|
|
%0 = getelementptr inbounds i32, i32* %s1, i32 %index
|
|
%1 = bitcast i32* %0 to <4 x i32>*
|
|
%wide.load = load <4 x i32>, <4 x i32>* %1, align 4
|
|
%2 = getelementptr inbounds i32, i32* %d, i32 %index
|
|
%3 = bitcast i32* %2 to <4 x i32>*
|
|
%wide.load10 = load <4 x i32>, <4 x i32>* %3, align 4
|
|
%4 = mul nsw <4 x i32> %wide.load, %wide.load10
|
|
%5 = add nsw <4 x i32> %broadcast.splat9, %4
|
|
%6 = bitcast i32* %2 to <4 x i32>*
|
|
store <4 x i32> %5, <4 x i32>* %6, align 4
|
|
%index.next = add i32 %index, 4
|
|
%7 = icmp eq i32 %index.next, %n
|
|
br i1 %7, label %for.cond.cleanup, label %vector.body
|
|
|
|
for.cond.cleanup:
|
|
ret void
|
|
}
|
|
|
|
define void @vmlas16_in_loop(i16* %s1, i16 %x, i16* %d, i32 %n) {
|
|
; CHECK-LABEL: vmlas16_in_loop:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: .LBB16_1: @ %vector.body
|
|
; CHECK-NEXT: @ =>This Inner Loop Header: Depth=1
|
|
; CHECK-NEXT: vldrh.u16 q0, [r2]
|
|
; CHECK-NEXT: vldrh.u16 q1, [r0], #16
|
|
; CHECK-NEXT: subs r3, #8
|
|
; CHECK-NEXT: vmlas.u16 q1, q0, r1
|
|
; CHECK-NEXT: vstrb.8 q1, [r2], #16
|
|
; CHECK-NEXT: bne .LBB16_1
|
|
; CHECK-NEXT: @ %bb.2: @ %for.cond.cleanup
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%broadcast.splatinsert11 = insertelement <8 x i16> undef, i16 %x, i32 0
|
|
%broadcast.splat12 = shufflevector <8 x i16> %broadcast.splatinsert11, <8 x i16> undef, <8 x i32> zeroinitializer
|
|
br label %vector.body
|
|
|
|
vector.body:
|
|
%index = phi i32 [ 0, %entry ], [ %index.next, %vector.body ]
|
|
%0 = getelementptr inbounds i16, i16* %s1, i32 %index
|
|
%1 = bitcast i16* %0 to <8 x i16>*
|
|
%wide.load = load <8 x i16>, <8 x i16>* %1, align 2
|
|
%2 = getelementptr inbounds i16, i16* %d, i32 %index
|
|
%3 = bitcast i16* %2 to <8 x i16>*
|
|
%wide.load13 = load <8 x i16>, <8 x i16>* %3, align 2
|
|
%4 = mul <8 x i16> %wide.load, %wide.load13
|
|
%5 = add <8 x i16> %4, %broadcast.splat12
|
|
%6 = bitcast i16* %2 to <8 x i16>*
|
|
store <8 x i16> %5, <8 x i16>* %6, align 2
|
|
%index.next = add i32 %index, 8
|
|
%7 = icmp eq i32 %index.next, %n
|
|
br i1 %7, label %for.cond.cleanup, label %vector.body
|
|
|
|
for.cond.cleanup:
|
|
ret void
|
|
}
|
|
|
|
define void @vmlas8_in_loop(i8* %s1, i8 %x, i8* %d, i32 %n) {
|
|
; CHECK-LABEL: vmlas8_in_loop:
|
|
; CHECK: @ %bb.0: @ %entry
|
|
; CHECK-NEXT: .LBB17_1: @ %vector.body
|
|
; CHECK-NEXT: @ =>This Inner Loop Header: Depth=1
|
|
; CHECK-NEXT: vldrh.u16 q0, [r2]
|
|
; CHECK-NEXT: vldrh.u16 q1, [r0], #16
|
|
; CHECK-NEXT: subs r3, #16
|
|
; CHECK-NEXT: vmlas.u8 q1, q0, r1
|
|
; CHECK-NEXT: vstrb.8 q1, [r2], #16
|
|
; CHECK-NEXT: bne .LBB17_1
|
|
; CHECK-NEXT: @ %bb.2: @ %for.cond.cleanup
|
|
; CHECK-NEXT: bx lr
|
|
entry:
|
|
%broadcast.splatinsert11 = insertelement <16 x i8> undef, i8 %x, i32 0
|
|
%broadcast.splat12 = shufflevector <16 x i8> %broadcast.splatinsert11, <16 x i8> undef, <16 x i32> zeroinitializer
|
|
br label %vector.body
|
|
|
|
vector.body:
|
|
%index = phi i32 [ 0, %entry ], [ %index.next, %vector.body ]
|
|
%0 = getelementptr inbounds i8, i8* %s1, i32 %index
|
|
%1 = bitcast i8* %0 to <16 x i8>*
|
|
%wide.load = load <16 x i8>, <16 x i8>* %1, align 2
|
|
%2 = getelementptr inbounds i8, i8* %d, i32 %index
|
|
%3 = bitcast i8* %2 to <16 x i8>*
|
|
%wide.load13 = load <16 x i8>, <16 x i8>* %3, align 2
|
|
%4 = mul <16 x i8> %wide.load, %wide.load13
|
|
%5 = add <16 x i8> %4, %broadcast.splat12
|
|
%6 = bitcast i8* %2 to <16 x i8>*
|
|
store <16 x i8> %5, <16 x i8>* %6, align 2
|
|
%index.next = add i32 %index, 16
|
|
%7 = icmp eq i32 %index.next, %n
|
|
br i1 %7, label %for.cond.cleanup, label %vector.body
|
|
|
|
for.cond.cleanup:
|
|
ret void
|
|
}
|