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 11module; 12export module A; 13 14export void f_streaming(void) __arm_streaming { } 15export void f_streaming_compatible(void) __arm_streaming_compatible { } 16export void f_shared_za(void) __arm_inout("za") { } 17export void f_preserves_za(void) __arm_preserves("za") { } 18 19//--- Use.cpp 20// expected-no-diagnostics 21import 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_inout_za" {{.*}}} 47// CHECK-DAG: attributes #[[SHARED_ZA_DECL]] = {{{.*}} "aarch64_inout_za" {{.*}}} 48// CHECK-DAG: attributes #[[PRESERVES_ZA_DECL]] = {{{.*}} "aarch64_preserves_za" {{.*}}} 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_inout_za" } 53// CHECK-DAG: attributes #[[PRESERVES_ZA_USE]] = { "aarch64_preserves_za" } 54// CHECK-DAG: attributes #[[STREAMING_USE]] = { "aarch64_pstate_sm_enabled" } 55// CHECK-DAG: attributes #[[STREAMING_COMPATIBLE_USE]] = { "aarch64_pstate_sm_compatible" } 56 57void f_shared_za_caller(void) __arm_inout("za") { 58 f_shared_za(); 59 f_preserves_za(); 60} 61 62void f_nonstreaming_caller(void) { 63 f_streaming(); 64 f_streaming_compatible(); 65} 66