1 // RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve2 -fallow-half-arguments-and-returns -S -O1 -Werror -Wall -emit-llvm -o - %s | FileCheck %s
2 // RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve2 -fallow-half-arguments-and-returns -S -O1 -Werror -Wall -emit-llvm -o - -x c++ %s | FileCheck %s
3 // RUN: %clang_cc1 -DSVE_OVERLOADED_FORMS -triple aarch64-none-linux-gnu -target-feature +sve2 -fallow-half-arguments-and-returns -S -O1 -Werror -Wall -emit-llvm -o - %s | FileCheck %s
4 // RUN: %clang_cc1 -DSVE_OVERLOADED_FORMS -triple aarch64-none-linux-gnu -target-feature +sve2 -fallow-half-arguments-and-returns -S -O1 -Werror -Wall -emit-llvm -o - -x c++ %s | FileCheck %s
5 // RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve -fallow-half-arguments-and-returns -fsyntax-only -verify -verify-ignore-unexpected=error %s
6 // RUN: %clang_cc1 -DSVE_OVERLOADED_FORMS -triple aarch64-none-linux-gnu -target-feature +sve -fallow-half-arguments-and-returns -fsyntax-only -verify=overload -verify-ignore-unexpected=error %s
7 
8 #include <arm_sve.h>
9 
10 #ifdef SVE_OVERLOADED_FORMS
11 // A simple used,unused... macro, long enough to represent any SVE builtin.
12 #define SVE_ACLE_FUNC(A1,A2_UNUSED,A3,A4_UNUSED) A1##A3
13 #else
14 #define SVE_ACLE_FUNC(A1,A2,A3,A4) A1##A2##A3##A4
15 #endif
16 
test_svtbx_s8(svint8_t fallback,svint8_t data,svuint8_t indices)17 svint8_t test_svtbx_s8(svint8_t fallback, svint8_t data, svuint8_t indices)
18 {
19   // CHECK-LABEL: test_svtbx_s8
20   // CHECK: %[[INTRINSIC:.*]] = call <vscale x 16 x i8> @llvm.aarch64.sve.tbx.nxv16i8(<vscale x 16 x i8> %fallback, <vscale x 16 x i8> %data, <vscale x 16 x i8> %indices)
21   // CHECK: ret <vscale x 16 x i8> %[[INTRINSIC]]
22   // overload-warning@+2 {{implicit declaration of function 'svtbx'}}
23   // expected-warning@+1 {{implicit declaration of function 'svtbx_s8'}}
24   return SVE_ACLE_FUNC(svtbx,_s8,,)(fallback, data, indices);
25 }
26 
test_svtbx_s16(svint16_t fallback,svint16_t data,svuint16_t indices)27 svint16_t test_svtbx_s16(svint16_t fallback, svint16_t data, svuint16_t indices)
28 {
29   // CHECK-LABEL: test_svtbx_s16
30   // CHECK: %[[INTRINSIC:.*]] = call <vscale x 8 x i16> @llvm.aarch64.sve.tbx.nxv8i16(<vscale x 8 x i16> %fallback, <vscale x 8 x i16> %data, <vscale x 8 x i16> %indices)
31   // CHECK: ret <vscale x 8 x i16> %[[INTRINSIC]]
32   // overload-warning@+2 {{implicit declaration of function 'svtbx'}}
33   // expected-warning@+1 {{implicit declaration of function 'svtbx_s16'}}
34   return SVE_ACLE_FUNC(svtbx,_s16,,)(fallback, data, indices);
35 }
36 
test_svtbx_s32(svint32_t fallback,svint32_t data,svuint32_t indices)37 svint32_t test_svtbx_s32(svint32_t fallback, svint32_t data, svuint32_t indices)
38 {
39   // CHECK-LABEL: test_svtbx_s32
40   // CHECK: %[[INTRINSIC:.*]] = call <vscale x 4 x i32> @llvm.aarch64.sve.tbx.nxv4i32(<vscale x 4 x i32> %fallback, <vscale x 4 x i32> %data, <vscale x 4 x i32> %indices)
41   // CHECK: ret <vscale x 4 x i32> %[[INTRINSIC]]
42   // overload-warning@+2 {{implicit declaration of function 'svtbx'}}
43   // expected-warning@+1 {{implicit declaration of function 'svtbx_s32'}}
44   return SVE_ACLE_FUNC(svtbx,_s32,,)(fallback, data, indices);
45 }
46 
test_svtbx_s64(svint64_t fallback,svint64_t data,svuint64_t indices)47 svint64_t test_svtbx_s64(svint64_t fallback, svint64_t data, svuint64_t indices)
48 {
49   // CHECK-LABEL: test_svtbx_s64
50   // CHECK: %[[INTRINSIC:.*]] = call <vscale x 2 x i64> @llvm.aarch64.sve.tbx.nxv2i64(<vscale x 2 x i64> %fallback, <vscale x 2 x i64> %data, <vscale x 2 x i64> %indices)
51   // CHECK: ret <vscale x 2 x i64> %[[INTRINSIC]]
52   // overload-warning@+2 {{implicit declaration of function 'svtbx'}}
53   // expected-warning@+1 {{implicit declaration of function 'svtbx_s64'}}
54   return SVE_ACLE_FUNC(svtbx,_s64,,)(fallback, data, indices);
55 }
56 
test_svtbx_u8(svuint8_t fallback,svuint8_t data,svuint8_t indices)57 svuint8_t test_svtbx_u8(svuint8_t fallback, svuint8_t data, svuint8_t indices)
58 {
59   // CHECK-LABEL: test_svtbx_u8
60   // CHECK: %[[INTRINSIC:.*]] = call <vscale x 16 x i8> @llvm.aarch64.sve.tbx.nxv16i8(<vscale x 16 x i8> %fallback, <vscale x 16 x i8> %data, <vscale x 16 x i8> %indices)
61   // CHECK: ret <vscale x 16 x i8> %[[INTRINSIC]]
62   // overload-warning@+2 {{implicit declaration of function 'svtbx'}}
63   // expected-warning@+1 {{implicit declaration of function 'svtbx_u8'}}
64   return SVE_ACLE_FUNC(svtbx,_u8,,)(fallback, data, indices);
65 }
66 
test_svtbx_u16(svuint16_t fallback,svuint16_t data,svuint16_t indices)67 svuint16_t test_svtbx_u16(svuint16_t fallback, svuint16_t data, svuint16_t indices)
68 {
69   // CHECK-LABEL: test_svtbx_u16
70   // CHECK: %[[INTRINSIC:.*]] = call <vscale x 8 x i16> @llvm.aarch64.sve.tbx.nxv8i16(<vscale x 8 x i16> %fallback, <vscale x 8 x i16> %data, <vscale x 8 x i16> %indices)
71   // CHECK: ret <vscale x 8 x i16> %[[INTRINSIC]]
72   // overload-warning@+2 {{implicit declaration of function 'svtbx'}}
73   // expected-warning@+1 {{implicit declaration of function 'svtbx_u16'}}
74   return SVE_ACLE_FUNC(svtbx,_u16,,)(fallback, data, indices);
75 }
76 
test_svtbx_u32(svuint32_t fallback,svuint32_t data,svuint32_t indices)77 svuint32_t test_svtbx_u32(svuint32_t fallback, svuint32_t data, svuint32_t indices)
78 {
79   // CHECK-LABEL: test_svtbx_u32
80   // CHECK: %[[INTRINSIC:.*]] = call <vscale x 4 x i32> @llvm.aarch64.sve.tbx.nxv4i32(<vscale x 4 x i32> %fallback, <vscale x 4 x i32> %data, <vscale x 4 x i32> %indices)
81   // CHECK: ret <vscale x 4 x i32> %[[INTRINSIC]]
82   // overload-warning@+2 {{implicit declaration of function 'svtbx'}}
83   // expected-warning@+1 {{implicit declaration of function 'svtbx_u32'}}
84   return SVE_ACLE_FUNC(svtbx,_u32,,)(fallback, data, indices);
85 }
86 
test_svtbx_u64(svuint64_t fallback,svuint64_t data,svuint64_t indices)87 svuint64_t test_svtbx_u64(svuint64_t fallback, svuint64_t data, svuint64_t indices)
88 {
89   // CHECK-LABEL: test_svtbx_u64
90   // CHECK: %[[INTRINSIC:.*]] = call <vscale x 2 x i64> @llvm.aarch64.sve.tbx.nxv2i64(<vscale x 2 x i64> %fallback, <vscale x 2 x i64> %data, <vscale x 2 x i64> %indices)
91   // CHECK: ret <vscale x 2 x i64> %[[INTRINSIC]]
92   // overload-warning@+2 {{implicit declaration of function 'svtbx'}}
93   // expected-warning@+1 {{implicit declaration of function 'svtbx_u64'}}
94   return SVE_ACLE_FUNC(svtbx,_u64,,)(fallback, data, indices);
95 }
96 
test_svtbx_f16(svfloat16_t fallback,svfloat16_t data,svuint16_t indices)97 svfloat16_t test_svtbx_f16(svfloat16_t fallback, svfloat16_t data, svuint16_t indices)
98 {
99   // CHECK-LABEL: test_svtbx_f16
100   // CHECK: %[[INTRINSIC:.*]] = call <vscale x 8 x half> @llvm.aarch64.sve.tbx.nxv8f16(<vscale x 8 x half> %fallback, <vscale x 8 x half> %data, <vscale x 8 x i16> %indices)
101   // CHECK: ret <vscale x 8 x half> %[[INTRINSIC]]
102   // overload-warning@+2 {{implicit declaration of function 'svtbx'}}
103   // expected-warning@+1 {{implicit declaration of function 'svtbx_f16'}}
104   return SVE_ACLE_FUNC(svtbx,_f16,,)(fallback, data, indices);
105 }
106 
test_svtbx_f32(svfloat32_t fallback,svfloat32_t data,svuint32_t indices)107 svfloat32_t test_svtbx_f32(svfloat32_t fallback, svfloat32_t data, svuint32_t indices)
108 {
109   // CHECK-LABEL: test_svtbx_f32
110   // CHECK: %[[INTRINSIC:.*]] = call <vscale x 4 x float> @llvm.aarch64.sve.tbx.nxv4f32(<vscale x 4 x float> %fallback, <vscale x 4 x float> %data, <vscale x 4 x i32> %indices)
111   // CHECK: ret <vscale x 4 x float> %[[INTRINSIC]]
112   // overload-warning@+2 {{implicit declaration of function 'svtbx'}}
113   // expected-warning@+1 {{implicit declaration of function 'svtbx_f32'}}
114   return SVE_ACLE_FUNC(svtbx,_f32,,)(fallback, data, indices);
115 }
116 
test_svtbx_f64(svfloat64_t fallback,svfloat64_t data,svuint64_t indices)117 svfloat64_t test_svtbx_f64(svfloat64_t fallback, svfloat64_t data, svuint64_t indices)
118 {
119   // CHECK-LABEL: test_svtbx_f64
120   // CHECK: %[[INTRINSIC:.*]] = call <vscale x 2 x double> @llvm.aarch64.sve.tbx.nxv2f64(<vscale x 2 x double> %fallback, <vscale x 2 x double> %data, <vscale x 2 x i64> %indices)
121   // CHECK: ret <vscale x 2 x double> %[[INTRINSIC]]
122   // overload-warning@+2 {{implicit declaration of function 'svtbx'}}
123   // expected-warning@+1 {{implicit declaration of function 'svtbx_f64'}}
124   return SVE_ACLE_FUNC(svtbx,_f64,,)(fallback, data, indices);
125 }
126