test/CodeGen/AArch64/sve-vector-splat.ll

   1 ; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve < %s | FileCheck %s
   2
   3 ;; Splats of legal integer vector types
   4
   5 define <vscale x 16 x i8> @sve_splat_16xi8(i8 %val) {
   6 ; CHECK-LABEL: @sve_splat_16xi8
   7 ; CHECK: mov z0.b, w0
   8 ; CHECK-NEXT: ret
   9   %ins = insertelement <vscale x 16 x i8> undef, i8 %val, i32 0
  10   %splat = shufflevector <vscale x 16 x i8> %ins, <vscale x 16 x i8> undef, <vscale x 16 x i32> zeroinitializer
  11   ret <vscale x 16 x i8> %splat
  12 }
  13
  14 define <vscale x 8 x i16> @sve_splat_8xi16(i16 %val) {
  15 ; CHECK-LABEL: @sve_splat_8xi16
  16 ; CHECK: mov z0.h, w0
  17 ; CHECK-NEXT: ret
  18   %ins = insertelement <vscale x 8 x i16> undef, i16 %val, i32 0
  19   %splat = shufflevector <vscale x 8 x i16> %ins, <vscale x 8 x i16> undef, <vscale x 8 x i32> zeroinitializer
  20   ret <vscale x 8 x i16> %splat
  21 }
  22
  23 define <vscale x 4 x i32> @sve_splat_4xi32(i32 %val) {
  24 ; CHECK-LABEL: @sve_splat_4xi32
  25 ; CHECK: mov z0.s, w0
  26 ; CHECK-NEXT: ret
  27   %ins = insertelement <vscale x 4 x i32> undef, i32 %val, i32 0
  28   %splat = shufflevector <vscale x 4 x i32> %ins, <vscale x 4 x i32> undef, <vscale x 4 x i32> zeroinitializer
  29   ret <vscale x 4 x i32> %splat
  30 }
  31
  32 define <vscale x 2 x i64> @sve_splat_2xi64(i64 %val) {
  33 ; CHECK-LABEL: @sve_splat_2xi64
  34 ; CHECK: mov z0.d, x0
  35 ; CHECK-NEXT: ret
  36   %ins = insertelement <vscale x 2 x i64> undef, i64 %val, i32 0
  37   %splat = shufflevector <vscale x 2 x i64> %ins, <vscale x 2 x i64> undef, <vscale x 2 x i32> zeroinitializer
  38   ret <vscale x 2 x i64> %splat
  39 }
  40
  41 ;; Promote splats of smaller illegal integer vector types
  42
  43 define <vscale x 2 x i8> @sve_splat_2xi8(i8 %val) {
  44 ; CHECK-LABEL: @sve_splat_2xi8
  45 ; CHECK: mov z0.d, x0
  46 ; CHECK-NEXT: ret
  47   %ins = insertelement <vscale x 2 x i8> undef, i8 %val, i32 0
  48   %splat = shufflevector <vscale x 2 x i8> %ins, <vscale x 2 x i8> undef, <vscale x 2 x i32> zeroinitializer
  49   ret <vscale x 2 x i8> %splat
  50 }
  51
  52 define <vscale x 4 x i8> @sve_splat_4xi8(i8 %val) {
  53 ; CHECK-LABEL: @sve_splat_4xi8
  54 ; CHECK: mov z0.s, w0
  55 ; CHECK-NEXT: ret
  56   %ins = insertelement <vscale x 4 x i8> undef, i8 %val, i32 0
  57   %splat = shufflevector <vscale x 4 x i8> %ins, <vscale x 4 x i8> undef, <vscale x 4 x i32> zeroinitializer
  58   ret <vscale x 4 x i8> %splat
  59 }
  60
  61 define <vscale x 8 x i8> @sve_splat_8xi8(i8 %val) {
  62 ; CHECK-LABEL: @sve_splat_8xi8
  63 ; CHECK: mov z0.h, w0
  64 ; CHECK-NEXT: ret
  65   %ins = insertelement <vscale x 8 x i8> undef, i8 %val, i32 0
  66   %splat = shufflevector <vscale x 8 x i8> %ins, <vscale x 8 x i8> undef, <vscale x 8 x i32> zeroinitializer
  67   ret <vscale x 8 x i8> %splat
  68 }
  69
  70 define <vscale x 2 x i16> @sve_splat_2xi16(i16 %val) {
  71 ; CHECK-LABEL: @sve_splat_2xi16
  72 ; CHECK: mov z0.d, x0
  73 ; CHECK-NEXT: ret
  74   %ins = insertelement <vscale x 2 x i16> undef, i16 %val, i32 0
  75   %splat = shufflevector <vscale x 2 x i16> %ins, <vscale x 2 x i16> undef, <vscale x 2 x i32> zeroinitializer
  76   ret <vscale x 2 x i16> %splat
  77 }
  78
  79 define <vscale x 4 x i16> @sve_splat_4xi16(i16 %val) {
  80 ; CHECK-LABEL: @sve_splat_4xi16
  81 ; CHECK: mov z0.s, w0
  82 ; CHECK-NEXT: ret
  83   %ins = insertelement <vscale x 4 x i16> undef, i16 %val, i32 0
  84   %splat = shufflevector <vscale x 4 x i16> %ins, <vscale x 4 x i16> undef, <vscale x 4 x i32> zeroinitializer
  85   ret <vscale x 4 x i16> %splat
  86 }
  87
  88 define <vscale x 2 x i32> @sve_splat_2xi32(i32 %val) {
  89 ; CHECK-LABEL: @sve_splat_2xi32
  90 ; CHECK: mov z0.d, x0
  91 ; CHECK-NEXT: ret
  92   %ins = insertelement <vscale x 2 x i32> undef, i32 %val, i32 0
  93   %splat = shufflevector <vscale x 2 x i32> %ins, <vscale x 2 x i32> undef, <vscale x 2 x i32> zeroinitializer
  94   ret <vscale x 2 x i32> %splat
  95 }