[llvm] a50037a - [AArch64] Add a few more vector extension tests.

[Florian Hahn via llvm-commits]

Author: Florian Hahn
Date: 2021-03-29T18:56:00+01:00
New Revision: a50037aaa6d5df403ff6b709a367164313ffd23f

URL: https://github.com/llvm/llvm-project/commit/a50037aaa6d5df403ff6b709a367164313ffd23f
DIFF: https://github.com/llvm/llvm-project/commit/a50037aaa6d5df403ff6b709a367164313ffd23f.diff

LOG: [AArch64] Add a few more vector extension tests.

Added: 
    

Modified: 
    llvm/test/CodeGen/AArch64/arm64-subvector-extend.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/CodeGen/AArch64/arm64-subvector-extend.ll b/llvm/test/CodeGen/AArch64/arm64-subvector-extend.ll
index e55ae25391f4..c0eefa145895 100644
--- a/llvm/test/CodeGen/AArch64/arm64-subvector-extend.ll
+++ b/llvm/test/CodeGen/AArch64/arm64-subvector-extend.ll
@@ -145,3 +145,60 @@ define <4 x i64> @cfunc4(<4 x i16> %v0) nounwind {
   %r = sext <4 x i16> %v0 to <4 x i64>
   ret <4 x i64> %r
 }
+
+define <4 x i64> @zext_v4i8_to_v4i64(<4 x i8> %v0) nounwind {
+; CHECK-LABEL: zext_v4i8_to_v4i64:
+; CHECK-NEXT:    bic.4h  v0, #255, lsl #8
+; CHECK-NEXT:    ushll.4s    v0, v0, #0
+; CHECK-NEXT:    ushll2.2d   v1, v0, #0
+; CHECK-NEXT:    ushll.2d    v0, v0, #0
+; CHECK-NEXT:    ret
+;
+  %r = zext <4 x i8> %v0 to <4 x i64>
+  ret <4 x i64> %r
+}
+
+define <4 x i64> @sext_v4i8_to_v4i64(<4 x i8> %v0) nounwind {
+; CHECK-LABEL: sext_v4i8_to_v4i64:
+; CHECK-NEXT:    ushll.4s    v0, v0, #0
+; CHECK-NEXT:    ushll.2d    v1, v0, #0
+; CHECK-NEXT:    ushll2.2d   v0, v0, #0
+; CHECK-NEXT:    shl.2d  v0, v0, #56
+; CHECK-NEXT:    shl.2d  v2, v1, #56
+; CHECK-NEXT:    sshr.2d v1, v0, #56
+; CHECK-NEXT:    sshr.2d v0, v2, #56
+; CHECK-NEXT:    ret
+;
+  %r = sext <4 x i8> %v0 to <4 x i64>
+  ret <4 x i64> %r
+}
+
+define <8 x i64> @zext_v8i8_to_v8i64(<8 x i8> %v0) nounwind {
+; CHECK-LABEL: zext_v8i8_to_v8i64:
+; CHECK-NEXT:    ushll.8h   v0, v0, #0
+; CHECK-NEXT:    ushll2.4s  v2, v0, #0
+; CHECK-NEXT:    ushll.4s   v0, v0, #0
+; CHECK-NEXT:    ushll2.2d  v3, v2, #0
+; CHECK-NEXT:    ushll2.2d  v1, v0, #0
+; CHECK-NEXT:    ushll.2d   v2, v2, #0
+; CHECK-NEXT:    ushll.2d   v0, v0, #0
+; CHECK-NEXT:    ret
+;
+  %r = zext <8 x i8> %v0 to <8 x i64>
+  ret <8 x i64> %r
+}
+
+define <8 x i64> @sext_v8i8_to_v8i64(<8 x i8> %v0) nounwind {
+; CHECK-LABEL: sext_v8i8_to_v8i64:
+; CHECK-NEXT:    sshll.8h   v0, v0, #0
+; CHECK-NEXT:    sshll2.4s  v2, v0, #0
+; CHECK-NEXT:    sshll.4s   v0, v0, #0
+; CHECK-NEXT:    sshll2.2d  v3, v2, #0
+; CHECK-NEXT:    sshll2.2d  v1, v0, #0
+; CHECK-NEXT:    sshll.2d   v2, v2, #0
+; CHECK-NEXT:    sshll.2d   v0, v0, #0
+; CHECK-NEXT:    ret
+;
+  %r = sext <8 x i8> %v0 to <8 x i64>
+  ret <8 x i64> %r
+}