[AArch64] Add a few more vector extension tests.

diff --git a/llvm/test/CodeGen/AArch64/arm64-subvector-extend.ll b/llvm/test/CodeGen/AArch64/arm64-subvector-extend.ll
index e55ae25..c0eefa1 100644 (file)
--- 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
+}