xref: /llvm-project/llvm/test/Analysis/CostModel/AArch64/cost-scalable-vector-gep.ll (revision 162bafc8b7a9f25f793bd58c7348e4c94b4a0a84)
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
8target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128"
9
10define 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
25define 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}
35