[Mlir-commits] [mlir] Add more ZA modes (PR #77361)

[Cullen Rhodes]

================
@@ -220,6 +220,27 @@ define void @streaming_compatible_func() "aarch64_pstate_sm_compatible" {
 
 // -----
 
+; CHECK-LABEL: @arm_new_za_func
+; CHECK-SAME: attributes {arm_new_za, arm_streaming}
+define void @arm_new_za_func() "aarch64_pstate_sm_enabled" "aarch64_pstate_za_new" {
+  ret void
+}
+
+
+; CHECK-LABEL: @arm_preserves_za_func
+; CHECK-SAME: attributes {arm_preserves_za, arm_streaming}
+define void @arm_preserves_za_func() "aarch64_pstate_sm_enabled" "aarch64_pstate_za_preserved" {
+  ret void
+}
+
+; CHECK-LABEL: @arm_shared_za_func
+; CHECK-SAME: attributes {arm_shared_za, arm_streaming}
+define void @arm_shared_za_func() "aarch64_pstate_sm_enabled" "aarch64_pstate_za_shared" {
+  ret void
+}
----------------
c-rhodes wrote:

for the purpose of these tests `"aarch64_pstate_sm_enabled"` isn't necessary

https://github.com/llvm/llvm-project/pull/77361