llvm/test/CodeGen/AArch64/scalable-vector-promotion.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt -mtriple=aarch64 -codegenprepare -S < %s | FileCheck %s
   3
   4 ; This test intends to check vector promotion for scalable vector. Current target lowering
   5 ; rejects scalable vector before reaching getConstantVector() in CodeGenPrepare. This test
   6 ; will assert once target lowering is ready, then we can bring in implementation for non-splat
   7 ; codepath for scalable vector.
   8
   9 define void @simpleOneInstructionPromotion(ptr %addr1, ptr %dest) {
  10 ; CHECK-LABEL: @simpleOneInstructionPromotion(
  11 ; CHECK-NEXT:    [[IN1:%.*]] = load <vscale x 2 x i32>, ptr [[ADDR1:%.*]], align 8
  12 ; CHECK-NEXT:    [[EXTRACT:%.*]] = extractelement <vscale x 2 x i32> [[IN1]], i32 1
  13 ; CHECK-NEXT:    [[OUT:%.*]] = or i32 [[EXTRACT]], 1
  14 ; CHECK-NEXT:    store i32 [[OUT]], ptr [[DEST:%.*]], align 4
  15 ; CHECK-NEXT:    ret void
  16 ;
  17   %in1 = load <vscale x 2 x i32>, ptr %addr1, align 8
  18   %extract = extractelement <vscale x 2 x i32> %in1, i32 1
  19   %out = or i32 %extract, 1
  20   store i32 %out, ptr %dest, align 4
  21   ret void
  22 }
  23