252
|
1 // REQUIRES: aarch64-registered-target
|
|
2 //
|
|
3 // RUN: rm -rf %t
|
|
4 // RUN: split-file %s %t
|
|
5 //
|
|
6 // RUN: %clang_cc1 -std=c++20 -triple aarch64 -target-feature +sme %t/A.cppm -emit-module-interface -o %t/A.pcm
|
|
7 // RUN: %clang_cc1 -std=c++20 -triple aarch64 -target-feature +sme -fprebuilt-module-path=%t -I%t %t/Use.cpp -emit-llvm
|
|
8 // RUN: cat %t/Use.ll | FileCheck %s
|
|
9
|
|
10 //--- A.cppm
|
|
11 module;
|
|
12 export module A;
|
|
13
|
|
14 export void f_streaming(void) __arm_streaming { }
|
|
15 export void f_streaming_compatible(void) __arm_streaming_compatible { }
|
|
16 export void f_shared_za(void) __arm_shared_za { }
|
|
17 export void f_preserves_za(void) __arm_preserves_za { }
|
|
18
|
|
19 //--- Use.cpp
|
|
20 // expected-no-diagnostics
|
|
21 import A;
|
|
22
|
|
23 // CHECK: define dso_local void @_Z18f_shared_za_callerv() #[[SHARED_ZA_DEF:[0-9]+]] {
|
|
24 // CHECK: entry:
|
|
25 // CHECK: call void @_ZW1A11f_shared_zav() #[[SHARED_ZA_USE:[0-9]+]]
|
|
26 // CHECK: call void @_ZW1A14f_preserves_zav() #[[PRESERVES_ZA_USE:[0-9]+]]
|
|
27 // CHECK: ret void
|
|
28 // CHECK: }
|
|
29 //
|
|
30 // CHECK:declare void @_ZW1A11f_shared_zav() #[[SHARED_ZA_DECL:[0-9]+]]
|
|
31 //
|
|
32 // CHECK:declare void @_ZW1A14f_preserves_zav() #[[PRESERVES_ZA_DECL:[0-9]+]]
|
|
33 //
|
|
34 // CHECK:; Function Attrs: mustprogress noinline nounwind optnone
|
|
35 // CHECK:define dso_local void @_Z21f_nonstreaming_callerv() #[[NORMAL_DEF:[0-9]+]] {
|
|
36 // CHECK:entry:
|
|
37 // CHECK: call void @_ZW1A11f_streamingv() #[[STREAMING_USE:[0-9]+]]
|
|
38 // CHECK: call void @_ZW1A22f_streaming_compatiblev() #[[STREAMING_COMPATIBLE_USE:[0-9]+]]
|
|
39 // CHECK: ret void
|
|
40 // CHECK:}
|
|
41 //
|
|
42 // CHECK:declare void @_ZW1A11f_streamingv() #[[STREAMING_DECL:[0-9]+]]
|
|
43 //
|
|
44 // CHECK:declare void @_ZW1A22f_streaming_compatiblev() #[[STREAMING_COMPATIBLE_DECL:[0-9]+]]
|
|
45 //
|
|
46 // CHECK-DAG: attributes #[[SHARED_ZA_DEF]] = {{{.*}} "aarch64_pstate_za_shared" {{.*}}}
|
|
47 // CHECK-DAG: attributes #[[SHARED_ZA_DECL]] = {{{.*}} "aarch64_pstate_za_shared" {{.*}}}
|
|
48 // CHECK-DAG: attributes #[[PRESERVES_ZA_DECL]] = {{{.*}} "aarch64_pstate_za_preserved" {{.*}}}
|
|
49 // CHECK-DAG: attributes #[[NORMAL_DEF]] = {{{.*}}}
|
|
50 // CHECK-DAG: attributes #[[STREAMING_DECL]] = {{{.*}} "aarch64_pstate_sm_enabled" {{.*}}}
|
|
51 // CHECK-DAG: attributes #[[STREAMING_COMPATIBLE_DECL]] = {{{.*}} "aarch64_pstate_sm_compatible" {{.*}}}
|
|
52 // CHECK-DAG: attributes #[[SHARED_ZA_USE]] = { "aarch64_pstate_za_shared" }
|
|
53 // CHECK-DAG: attributes #[[PRESERVES_ZA_USE]] = { "aarch64_pstate_za_preserved" }
|
|
54 // CHECK-DAG: attributes #[[STREAMING_USE]] = { "aarch64_pstate_sm_enabled" }
|
|
55 // CHECK-DAG: attributes #[[STREAMING_COMPATIBLE_USE]] = { "aarch64_pstate_sm_compatible" }
|
|
56
|
|
57 void f_shared_za_caller(void) __arm_shared_za {
|
|
58 f_shared_za();
|
|
59 f_preserves_za();
|
|
60 }
|
|
61
|
|
62 void f_nonstreaming_caller(void) {
|
|
63 f_streaming();
|
|
64 f_streaming_compatible();
|
|
65 }
|