llvm/test/CodeGen/AArch64/sve-streaming-mode-fixed-length-build-vector.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc -mattr=+sve -force-streaming-compatible-sve  < %s | FileCheck %s
   3 ; RUN: llc -mattr=+sme -force-streaming-compatible-sve  < %s | FileCheck %s
   4
   5 target triple = "aarch64-unknown-linux-gnu"
   6
   7 define void @build_vector_7_inc1_v4i1(ptr %a) {
   8 ; CHECK-LABEL: build_vector_7_inc1_v4i1:
   9 ; CHECK:       // %bb.0:
  10 ; CHECK-NEXT:    mov w8, #5 // =0x5
  11 ; CHECK-NEXT:    strb w8, [x0]
  12 ; CHECK-NEXT:    ret
  13   store <4 x i1> <i1 true, i1 false, i1 true, i1 false>, ptr %a, align 1
  14   ret void
  15 }
  16
  17 define void @build_vector_7_inc1_v32i8(ptr %a) {
  18 ; CHECK-LABEL: build_vector_7_inc1_v32i8:
  19 ; CHECK:       // %bb.0:
  20 ; CHECK-NEXT:    index z0.b, #0, #1
  21 ; CHECK-NEXT:    mov z1.d, z0.d
  22 ; CHECK-NEXT:    add z0.b, z0.b, #7 // =0x7
  23 ; CHECK-NEXT:    add z1.b, z1.b, #23 // =0x17
  24 ; CHECK-NEXT:    stp q0, q1, [x0]
  25 ; CHECK-NEXT:    ret
  26   store <32 x i8> <i8 7, i8 8, i8 9, i8 10, i8 11, i8 12, i8 13, i8 14, i8 15, i8 16, i8 17, i8 18, i8 19, i8 20, i8 21, i8 22, i8 23, i8 24, i8 25, i8 26, i8 27, i8 28, i8 29, i8 30, i8 31, i8 32, i8 33, i8 34, i8 35, i8 36, i8 37, i8 38>, ptr %a, align 1
  27   ret void
  28 }
  29
  30 define void @build_vector_0_inc2_v16i16(ptr %a) {
  31 ; CHECK-LABEL: build_vector_0_inc2_v16i16:
  32 ; CHECK:       // %bb.0:
  33 ; CHECK-NEXT:    index z0.h, #0, #2
  34 ; CHECK-NEXT:    str q0, [x0]
  35 ; CHECK-NEXT:    add z0.h, z0.h, #16 // =0x10
  36 ; CHECK-NEXT:    str q0, [x0, #16]
  37 ; CHECK-NEXT:    ret
  38   store <16 x i16> <i16 0, i16 2, i16 4, i16 6, i16 8, i16 10, i16 12, i16 14, i16 16, i16 18, i16 20, i16 22, i16 24, i16 26, i16 28, i16 30>, ptr %a, align 2
  39   ret void
  40 }
  41
  42 ; Negative const stride.
  43 define void @build_vector_0_dec3_v8i32(ptr %a) {
  44 ; CHECK-LABEL: build_vector_0_dec3_v8i32:
  45 ; CHECK:       // %bb.0:
  46 ; CHECK-NEXT:    index z0.s, #0, #-3
  47 ; CHECK-NEXT:    mov z1.s, #-12 // =0xfffffffffffffff4
  48 ; CHECK-NEXT:    add z1.s, z0.s, z1.s
  49 ; CHECK-NEXT:    stp q0, q1, [x0]
  50 ; CHECK-NEXT:    ret
  51   store <8 x i32> <i32 0, i32 -3, i32 -6, i32 -9, i32 -12, i32 -15, i32 -18, i32 -21>, ptr %a, align 4
  52   ret void
  53 }
  54
  55 ; Constant stride that's too big to be directly encoded into the index.
  56 define void @build_vector_minus2_dec32_v4i64(ptr %a) {
  57 ; CHECK-LABEL: build_vector_minus2_dec32_v4i64:
  58 ; CHECK:       // %bb.0:
  59 ; CHECK-NEXT:    mov x8, #-32 // =0xffffffffffffffe0
  60 ; CHECK-NEXT:    mov z1.d, #-66 // =0xffffffffffffffbe
  61 ; CHECK-NEXT:    mov z2.d, #-2 // =0xfffffffffffffffe
  62 ; CHECK-NEXT:    index z0.d, #0, x8
  63 ; CHECK-NEXT:    add z1.d, z0.d, z1.d
  64 ; CHECK-NEXT:    add z0.d, z0.d, z2.d
  65 ; CHECK-NEXT:    stp q0, q1, [x0]
  66 ; CHECK-NEXT:    ret
  67   store <4 x i64> <i64 -2, i64 -34, i64 -66, i64 -98>, ptr %a, align 8
  68   ret void
  69 }
  70
  71 ; Constant but not a sequence.
  72 define void @build_vector_no_stride_v4i64(ptr %a) {
  73 ; CHECK-LABEL: build_vector_no_stride_v4i64:
  74 ; CHECK:       // %bb.0:
  75 ; CHECK-NEXT:    index z0.d, #1, #7
  76 ; CHECK-NEXT:    index z1.d, #0, #4
  77 ; CHECK-NEXT:    stp q1, q0, [x0]
  78 ; CHECK-NEXT:    ret
  79   store <4 x i64> <i64 0, i64 4, i64 1, i64 8>, ptr %a, align 8
  80   ret void
  81 }
  82
  83 define void @build_vector_0_inc2_v16f16(ptr %a) {
  84 ; CHECK-LABEL: build_vector_0_inc2_v16f16:
  85 ; CHECK:       // %bb.0:
  86 ; CHECK-NEXT:    adrp x8, .LCPI6_0
  87 ; CHECK-NEXT:    adrp x9, .LCPI6_1
  88 ; CHECK-NEXT:    ldr q0, [x8, :lo12:.LCPI6_0]
  89 ; CHECK-NEXT:    ldr q1, [x9, :lo12:.LCPI6_1]
  90 ; CHECK-NEXT:    stp q1, q0, [x0]
  91 ; CHECK-NEXT:    ret
  92   store <16 x half> <half 0.0, half 2.0, half 4.0, half 6.0, half 8.0, half 10.0, half 12.0, half 14.0, half 16.0, half 18.0, half 20.0, half 22.0, half 24.0, half 26.0, half 28.0, half 30.0>, ptr %a, align 2
  93   ret void
  94 }
  95
  96 ; Negative const stride.
  97 define void @build_vector_0_dec3_v8f32(ptr %a) {
  98 ; CHECK-LABEL: build_vector_0_dec3_v8f32:
  99 ; CHECK:       // %bb.0:
 100 ; CHECK-NEXT:    adrp x8, .LCPI7_0
 101 ; CHECK-NEXT:    adrp x9, .LCPI7_1
 102 ; CHECK-NEXT:    ldr q0, [x8, :lo12:.LCPI7_0]
 103 ; CHECK-NEXT:    ldr q1, [x9, :lo12:.LCPI7_1]
 104 ; CHECK-NEXT:    stp q1, q0, [x0]
 105 ; CHECK-NEXT:    ret
 106   store <8 x float> <float 0.0, float -3.0, float -6.0, float -9.0, float -12.0, float -15.0, float -18.0, float -21.0>, ptr %a, align 4
 107   ret void
 108 }
 109
 110 ; Constant stride that's too big to be directly encoded into the index.
 111 define void @build_vector_minus2_dec32_v4f64(ptr %a) {
 112 ; CHECK-LABEL: build_vector_minus2_dec32_v4f64:
 113 ; CHECK:       // %bb.0:
 114 ; CHECK-NEXT:    adrp x8, .LCPI8_0
 115 ; CHECK-NEXT:    adrp x9, .LCPI8_1
 116 ; CHECK-NEXT:    ldr q0, [x8, :lo12:.LCPI8_0]
 117 ; CHECK-NEXT:    ldr q1, [x9, :lo12:.LCPI8_1]
 118 ; CHECK-NEXT:    stp q1, q0, [x0]
 119 ; CHECK-NEXT:    ret
 120   store <4 x double> <double -2.0, double -34.0, double -66.0, double -98.0>, ptr %a, align 8
 121   ret void
 122 }
 123
 124 ; Constant but not a sequence.
 125 define void @build_vector_no_stride_v4f64(ptr %a) {
 126 ; CHECK-LABEL: build_vector_no_stride_v4f64:
 127 ; CHECK:       // %bb.0:
 128 ; CHECK-NEXT:    adrp x8, .LCPI9_0
 129 ; CHECK-NEXT:    adrp x9, .LCPI9_1
 130 ; CHECK-NEXT:    ldr q0, [x8, :lo12:.LCPI9_0]
 131 ; CHECK-NEXT:    ldr q1, [x9, :lo12:.LCPI9_1]
 132 ; CHECK-NEXT:    stp q1, q0, [x0]
 133 ; CHECK-NEXT:    ret
 134   store <4 x double> <double 0.0, double 4.0, double 1.0, double 8.0>, ptr %a, align 8
 135   ret void
 136 }