clang/test/Modules/aarch64-sme-keywords.cppm

   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_inout("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_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
  57 void f_shared_za_caller(void) __arm_inout("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 }