1 // RUN: %clang_cc1 -triple armv8.6a-arm-none-eabi -target-abi aapcs -mfloat-abi hard -target-feature +bf16 -target-feature +neon -emit-llvm -O2 -o - %s | opt -S -mem2reg -sroa | FileCheck %s --check-prefix=CHECK32-HARD
2 // RUN: %clang_cc1 -triple armv8.6a-arm-none-eabi -target-abi aapcs -mfloat-abi softfp -target-feature +bf16 -target-feature +neon -emit-llvm -O2 -o - %s | opt -S -mem2reg -sroa | FileCheck %s --check-prefix=CHECK32-SOFTFP
3 // RUN: %clang_cc1 -triple aarch64-arm-none-eabi -target-abi aapcs -target-feature +bf16 -target-feature +neon -emit-llvm -O2 -o - %s | opt -S -mem2reg -sroa | FileCheck %s --check-prefix=CHECK64
4 
5 #include <arm_neon.h>
6 
7 // function return types
8 __bf16 test_ret_bf16(__bf16 v) {
9   return v;
10 }
11 // CHECK32-HARD: define{{.*}} arm_aapcs_vfpcc bfloat @test_ret_bf16(bfloat returned %v) {{.*}} {
12 // CHECK32-HARD: ret bfloat %v
13 // CHECK32-SOFTFP: define{{.*}} bfloat @test_ret_bf16(bfloat returned %v) {{.*}} {
14 // CHECK32-SOFTFP: ret bfloat %v
15 // CHECK64: define{{.*}} bfloat @test_ret_bf16(bfloat returned %v) {{.*}} {
16 // CHECK64: ret bfloat %v
17 
18 bfloat16x4_t test_ret_bf16x4_t(bfloat16x4_t v) {
19   return v;
20 }
21 // CHECK32-HARD: define{{.*}} arm_aapcs_vfpcc <4 x bfloat> @test_ret_bf16x4_t(<4 x bfloat> returned %v) {{.*}} {
22 // CHECK32-HARD: ret <4 x bfloat> %v
23 // CHECK32-SOFTFP: define{{.*}} <2 x i32> @test_ret_bf16x4_t(<2 x i32> [[V0:.*]]) {{.*}} {
24 // CHECK32-SOFTFP: ret <2 x i32> %v
25 // CHECK64: define{{.*}} <4 x bfloat> @test_ret_bf16x4_t(<4 x bfloat> returned %v) {{.*}} {
26 // CHECK64: ret <4 x bfloat> %v
27