[CodeGen] Use SPLAT_VECTOR for zeroinitialiser with scalable types

david-arm · david-arm · commit 121ca44c19bf · 2020-04-27T16:16:48.000+01:00
Adding tests that I forgot to add as part of a previous change: https://reviews.llvm.org/D78636
diff --git a/llvm/test/CodeGen/AArch64/sve-zeroinit.ll b/llvm/test/CodeGen/AArch64/sve-zeroinit.ll
@@ -0,0 +1,81 @@
+; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve < %s | FileCheck %s
+
+target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
+target triple = "aarch64-none-linux-gnu"
+
+define <vscale x 2 x i64> @test_zeroinit_2xi64() {
+; CHECK-LABEL: test_zeroinit_2xi64
+; CHECK:       mov z0.d, #0
+; CHECK-NEXT:  ret
+  ret <vscale x 2 x i64> zeroinitializer
+}
+
+define <vscale x 4 x i32> @test_zeroinit_4xi32() {
+; CHECK-LABEL: test_zeroinit_4xi32
+; CHECK:       mov z0.s, #0
+; CHECK-NEXT:  ret
+  ret <vscale x 4 x i32> zeroinitializer
+}
+
+define <vscale x 8 x i16> @test_zeroinit_8xi16() {
+; CHECK-LABEL: test_zeroinit_8xi16
+; CHECK:       mov z0.h, #0
+; CHECK-NEXT:  ret
+  ret <vscale x 8 x i16> zeroinitializer
+}
+
+define <vscale x 16 x i8> @test_zeroinit_16xi8() {
+; CHECK-LABEL: test_zeroinit_16xi8
+; CHECK:       mov z0.b, #0
+; CHECK-NEXT:  ret
+  ret <vscale x 16 x i8> zeroinitializer
+}
+
+define <vscale x 2 x double> @test_zeroinit_2xf64() {
+; CHECK-LABEL: test_zeroinit_2xf64
+; CHECK:       mov z0.d, #0
+; CHECK-NEXT:  ret
+  ret <vscale x 2 x double> zeroinitializer
+}
+
+define <vscale x 4 x float> @test_zeroinit_4xf32() {
+; CHECK-LABEL: test_zeroinit_4xf32
+; CHECK:       mov z0.s, #0
+; CHECK-NEXT:  ret
+  ret <vscale x 4 x float> zeroinitializer
+}
+
+define <vscale x 8 x half> @test_zeroinit_8xf16() {
+; CHECK-LABEL: test_zeroinit_8xf16
+; CHECK:       mov z0.h, #0
+; CHECK-NEXT:  ret
+  ret <vscale x 8 x half> zeroinitializer
+}
+
+define <vscale x 2 x i1> @test_zeroinit_2xi1() {
+; CHECK-LABEL: test_zeroinit_2xi1
+; CHECK:       whilelo p0.d, xzr, xzr
+; CHECK-NEXT:  ret
+  ret <vscale x 2 x i1> zeroinitializer
+}
+
+define <vscale x 4 x i1> @test_zeroinit_4xi1() {
+; CHECK-LABEL: test_zeroinit_4xi1
+; CHECK:       whilelo p0.s, xzr, xzr
+; CHECK-NEXT:  ret
+  ret <vscale x 4 x i1> zeroinitializer
+}
+
+define <vscale x 8 x i1> @test_zeroinit_8xi1() {
+; CHECK-LABEL: test_zeroinit_8xi1
+; CHECK:       whilelo p0.h, xzr, xzr
+; CHECK-NEXT:  ret
+  ret <vscale x 8 x i1> zeroinitializer
+}
+
+define <vscale x 16 x i1> @test_zeroinit_16xi1() {
+; CHECK-LABEL: test_zeroinit_16xi1
+; CHECK:       whilelo p0.b, xzr, xzr
+; CHECK-NEXT:  ret
+  ret <vscale x 16 x i1> zeroinitializer
+}
