llvm/test/Transforms/InferAlignment/vscale.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
   2 ; RUN: opt -passes=infer-alignment -S < %s | FileCheck %s
   3
   4 ; <4 x i32> -> 16 byte alignment
   5 define void @alignment_sustain(ptr align 16 %ptr) {
   6 ; CHECK-LABEL: define void @alignment_sustain
   7 ; CHECK-SAME: (ptr align 16 [[PTR:%.*]]) {
   8 ; CHECK-NEXT:    [[GEP:%.*]] = getelementptr <vscale x 4 x i32>, ptr [[PTR]], i32 3
   9 ; CHECK-NEXT:    [[LOAD:%.*]] = load <4 x i32>, ptr [[GEP]], align 16
  10 ; CHECK-NEXT:    store <4 x i32> zeroinitializer, ptr [[GEP]], align 16
  11 ; CHECK-NEXT:    ret void
  12 ;
  13   %gep = getelementptr <vscale x 4 x i32>, ptr %ptr, i32 3
  14
  15   %load = load <4 x i32>, ptr %gep, align 16
  16   store <4 x i32> zeroinitializer, ptr %gep, align 16
  17
  18   ret void
  19 }
  20
  21 ; <8 x i32> -> 32 byte alignment
  22 define void @alignment_increase(ptr align 32 %ptr) {
  23 ; CHECK-LABEL: define void @alignment_increase
  24 ; CHECK-SAME: (ptr align 32 [[PTR:%.*]]) {
  25 ; CHECK-NEXT:    [[GEP:%.*]] = getelementptr <vscale x 8 x i32>, ptr [[PTR]], i32 3
  26 ; CHECK-NEXT:    [[LOAD:%.*]] = load <8 x i32>, ptr [[GEP]], align 32
  27 ; CHECK-NEXT:    store <8 x i32> zeroinitializer, ptr [[GEP]], align 32
  28 ; CHECK-NEXT:    ret void
  29 ;
  30   %gep = getelementptr <vscale x 8 x i32>, ptr %ptr, i32 3
  31
  32   %load = load <8 x i32>, ptr %gep, align 16
  33   store <8 x i32> zeroinitializer, ptr %gep, align 16
  34
  35   ret void
  36 }