1 // REQUIRES: aarch64-registered-target
2 // RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve -fallow-half-arguments-and-returns -S -O1 -Werror -Wall -emit-llvm -o - %s | FileCheck %s
3 // RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve -fallow-half-arguments-and-returns -S -O1 -Werror -Wall -emit-llvm -o - -x c++ %s | FileCheck %s
4 // RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve -fallow-half-arguments-and-returns -S -O1 -Werror -o - %s >/dev/null
5 #include <arm_sve.h>
6
test_svldnf1sh_s32(svbool_t pg,const int16_t * base)7 svint32_t test_svldnf1sh_s32(svbool_t pg, const int16_t *base)
8 {
9 // CHECK-LABEL: test_svldnf1sh_s32
10 // CHECK: %[[PG:.*]] = call <vscale x 4 x i1> @llvm.aarch64.sve.convert.from.svbool.nxv4i1(<vscale x 16 x i1> %pg)
11 // CHECK: %[[LOAD:.*]] = call <vscale x 4 x i16> @llvm.aarch64.sve.ldnf1.nxv4i16(<vscale x 4 x i1> %[[PG]], i16* %base)
12 // CHECK: %[[SEXT:.*]] = sext <vscale x 4 x i16> %[[LOAD]] to <vscale x 4 x i32>
13 // CHECK: ret <vscale x 4 x i32> %[[SEXT]]
14 return svldnf1sh_s32(pg, base);
15 }
16
test_svldnf1sh_s64(svbool_t pg,const int16_t * base)17 svint64_t test_svldnf1sh_s64(svbool_t pg, const int16_t *base)
18 {
19 // CHECK-LABEL: test_svldnf1sh_s64
20 // CHECK: %[[PG:.*]] = call <vscale x 2 x i1> @llvm.aarch64.sve.convert.from.svbool.nxv2i1(<vscale x 16 x i1> %pg)
21 // CHECK: %[[LOAD:.*]] = call <vscale x 2 x i16> @llvm.aarch64.sve.ldnf1.nxv2i16(<vscale x 2 x i1> %[[PG]], i16* %base)
22 // CHECK: %[[SEXT:.*]] = sext <vscale x 2 x i16> %[[LOAD]] to <vscale x 2 x i64>
23 // CHECK: ret <vscale x 2 x i64> %[[SEXT]]
24 return svldnf1sh_s64(pg, base);
25 }
26
test_svldnf1sh_u32(svbool_t pg,const int16_t * base)27 svuint32_t test_svldnf1sh_u32(svbool_t pg, const int16_t *base)
28 {
29 // CHECK-LABEL: test_svldnf1sh_u32
30 // CHECK: %[[PG:.*]] = call <vscale x 4 x i1> @llvm.aarch64.sve.convert.from.svbool.nxv4i1(<vscale x 16 x i1> %pg)
31 // CHECK: %[[LOAD:.*]] = call <vscale x 4 x i16> @llvm.aarch64.sve.ldnf1.nxv4i16(<vscale x 4 x i1> %[[PG]], i16* %base)
32 // CHECK: %[[SEXT:.*]] = sext <vscale x 4 x i16> %[[LOAD]] to <vscale x 4 x i32>
33 // CHECK: ret <vscale x 4 x i32> %[[SEXT]]
34 return svldnf1sh_u32(pg, base);
35 }
36
test_svldnf1sh_u64(svbool_t pg,const int16_t * base)37 svuint64_t test_svldnf1sh_u64(svbool_t pg, const int16_t *base)
38 {
39 // CHECK-LABEL: test_svldnf1sh_u64
40 // CHECK: %[[PG:.*]] = call <vscale x 2 x i1> @llvm.aarch64.sve.convert.from.svbool.nxv2i1(<vscale x 16 x i1> %pg)
41 // CHECK: %[[LOAD:.*]] = call <vscale x 2 x i16> @llvm.aarch64.sve.ldnf1.nxv2i16(<vscale x 2 x i1> %[[PG]], i16* %base)
42 // CHECK: %[[SEXT:.*]] = sext <vscale x 2 x i16> %[[LOAD]] to <vscale x 2 x i64>
43 // CHECK: ret <vscale x 2 x i64> %[[SEXT]]
44 return svldnf1sh_u64(pg, base);
45 }
46
test_svldnf1sh_vnum_s32(svbool_t pg,const int16_t * base,int64_t vnum)47 svint32_t test_svldnf1sh_vnum_s32(svbool_t pg, const int16_t *base, int64_t vnum)
48 {
49 // CHECK-LABEL: test_svldnf1sh_vnum_s32
50 // CHECK-DAG: %[[PG:.*]] = call <vscale x 4 x i1> @llvm.aarch64.sve.convert.from.svbool.nxv4i1(<vscale x 16 x i1> %pg)
51 // CHECK-DAG: %[[BITCAST:.*]] = bitcast i16* %base to <vscale x 4 x i16>*
52 // CHECK-DAG: %[[GEP:.*]] = getelementptr <vscale x 4 x i16>, <vscale x 4 x i16>* %[[BITCAST]], i64 %vnum, i64 0
53 // CHECK: %[[LOAD:.*]] = call <vscale x 4 x i16> @llvm.aarch64.sve.ldnf1.nxv4i16(<vscale x 4 x i1> %[[PG]], i16* %[[GEP]])
54 // CHECK: %[[SEXT:.*]] = sext <vscale x 4 x i16> %[[LOAD]] to <vscale x 4 x i32>
55 // CHECK: ret <vscale x 4 x i32> %[[SEXT]]
56 return svldnf1sh_vnum_s32(pg, base, vnum);
57 }
58
test_svldnf1sh_vnum_s64(svbool_t pg,const int16_t * base,int64_t vnum)59 svint64_t test_svldnf1sh_vnum_s64(svbool_t pg, const int16_t *base, int64_t vnum)
60 {
61 // CHECK-LABEL: test_svldnf1sh_vnum_s64
62 // CHECK-DAG: %[[PG:.*]] = call <vscale x 2 x i1> @llvm.aarch64.sve.convert.from.svbool.nxv2i1(<vscale x 16 x i1> %pg)
63 // CHECK-DAG: %[[BITCAST:.*]] = bitcast i16* %base to <vscale x 2 x i16>*
64 // CHECK-DAG: %[[GEP:.*]] = getelementptr <vscale x 2 x i16>, <vscale x 2 x i16>* %[[BITCAST]], i64 %vnum, i64 0
65 // CHECK: %[[LOAD:.*]] = call <vscale x 2 x i16> @llvm.aarch64.sve.ldnf1.nxv2i16(<vscale x 2 x i1> %[[PG]], i16* %[[GEP]])
66 // CHECK: %[[SEXT:.*]] = sext <vscale x 2 x i16> %[[LOAD]] to <vscale x 2 x i64>
67 // CHECK: ret <vscale x 2 x i64> %[[SEXT]]
68 return svldnf1sh_vnum_s64(pg, base, vnum);
69 }
70
test_svldnf1sh_vnum_u32(svbool_t pg,const int16_t * base,int64_t vnum)71 svuint32_t test_svldnf1sh_vnum_u32(svbool_t pg, const int16_t *base, int64_t vnum)
72 {
73 // CHECK-LABEL: test_svldnf1sh_vnum_u32
74 // CHECK-DAG: %[[PG:.*]] = call <vscale x 4 x i1> @llvm.aarch64.sve.convert.from.svbool.nxv4i1(<vscale x 16 x i1> %pg)
75 // CHECK-DAG: %[[BITCAST:.*]] = bitcast i16* %base to <vscale x 4 x i16>*
76 // CHECK-DAG: %[[GEP:.*]] = getelementptr <vscale x 4 x i16>, <vscale x 4 x i16>* %[[BITCAST]], i64 %vnum, i64 0
77 // CHECK: %[[LOAD:.*]] = call <vscale x 4 x i16> @llvm.aarch64.sve.ldnf1.nxv4i16(<vscale x 4 x i1> %[[PG]], i16* %[[GEP]])
78 // CHECK: %[[SEXT:.*]] = sext <vscale x 4 x i16> %[[LOAD]] to <vscale x 4 x i32>
79 // CHECK: ret <vscale x 4 x i32> %[[SEXT]]
80 return svldnf1sh_vnum_u32(pg, base, vnum);
81 }
82
test_svldnf1sh_vnum_u64(svbool_t pg,const int16_t * base,int64_t vnum)83 svuint64_t test_svldnf1sh_vnum_u64(svbool_t pg, const int16_t *base, int64_t vnum)
84 {
85 // CHECK-LABEL: test_svldnf1sh_vnum_u64
86 // CHECK-DAG: %[[PG:.*]] = call <vscale x 2 x i1> @llvm.aarch64.sve.convert.from.svbool.nxv2i1(<vscale x 16 x i1> %pg)
87 // CHECK-DAG: %[[BITCAST:.*]] = bitcast i16* %base to <vscale x 2 x i16>*
88 // CHECK-DAG: %[[GEP:.*]] = getelementptr <vscale x 2 x i16>, <vscale x 2 x i16>* %[[BITCAST]], i64 %vnum, i64 0
89 // CHECK: %[[LOAD:.*]] = call <vscale x 2 x i16> @llvm.aarch64.sve.ldnf1.nxv2i16(<vscale x 2 x i1> %[[PG]], i16* %[[GEP]])
90 // CHECK: %[[SEXT:.*]] = sext <vscale x 2 x i16> %[[LOAD]] to <vscale x 2 x i64>
91 // CHECK: ret <vscale x 2 x i64> %[[SEXT]]
92 return svldnf1sh_vnum_u64(pg, base, vnum);
93 }
94