llvm/test/Analysis/CostModel/AArch64/cost-scalable-vector-gep.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_analyze_test_checks.py UTC_ARGS: --version 2
   2 ; RUN: opt -passes="print<cost-model>" 2>&1 -disable-output -mtriple=aarch64--linux-gnu -mattr=+sve < %s | FileCheck %s
   3
   4 ; This regression test is verifying that a GEP instruction performed on a
   5 ; scalable vector does not produce a 'assumption that TypeSize is not scalable'
   6 ; warning when performing cost analysis.
   7
   8 target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
   9
  10 define void @gep_scalable_types(ptr %ptr) {
  11 ; CHECK-LABEL: 'gep_scalable_types'
  12 ; CHECK-NEXT:  Cost Model: Found an estimated cost of 1 for instruction: %gep1 = getelementptr <vscale x 16 x i8>, ptr %ptr, i32 2
  13 ; CHECK-NEXT:  Cost Model: Found an estimated cost of 1 for instruction: %gep2 = getelementptr [2 x <vscale x 16 x i8>], ptr %ptr, i32 2
  14 ; CHECK-NEXT:  Cost Model: Found an estimated cost of 1 for instruction: %gep3 = getelementptr target("aarch64.svcount"), ptr %ptr, i32 2
  15 ; CHECK-NEXT:  Cost Model: Found an estimated cost of 1 for instruction: %gep4 = getelementptr [2 x target("aarch64.svcount")], ptr %ptr, i32 2
  16 ; CHECK-NEXT:  Cost Model: Found an estimated cost of 0 for instruction: ret void
  17 ;
  18   %gep1 = getelementptr <vscale x 16 x i8>, ptr %ptr, i32 2
  19   %gep2 = getelementptr [2 x <vscale x 16 x i8>], ptr %ptr, i32 2
  20   %gep3 = getelementptr target("aarch64.svcount"), ptr %ptr, i32 2
  21   %gep4 = getelementptr [2 x target("aarch64.svcount")], ptr %ptr, i32 2
  22   ret void
  23 }
  24
  25 define ptr @sext_gep(ptr %p, i32 %a) {
  26 ; CHECK-LABEL: 'sext_gep'
  27 ; CHECK-NEXT:  Cost Model: Found an estimated cost of 1 for instruction: %b = sext i32 %a to i64
  28 ; CHECK-NEXT:  Cost Model: Found an estimated cost of 1 for instruction: %r = getelementptr <vscale x 8 x half>, ptr %p, i64 %b
  29 ; CHECK-NEXT:  Cost Model: Found an estimated cost of 0 for instruction: ret ptr %r
  30 ;
  31   %b = sext i32 %a to i64
  32   %r = getelementptr <vscale x 8 x half>, ptr %p, i64 %b
  33   ret ptr %r
  34 }