@@ -6334,8 +6334,10 @@ aarch64_function_ok_for_sibcall (tree, tree exp)
tree fntype = TREE_TYPE (TREE_TYPE (CALL_EXPR_FN (exp)));
if (aarch64_fntype_pstate_sm (fntype) & ~aarch64_cfun_incoming_pstate_sm ())
return false;
- if (aarch64_fntype_pstate_za (fntype) != aarch64_cfun_incoming_pstate_za ())
- return false;
+ for (auto state : { "za", "zt0" })
+ if (bool (aarch64_cfun_shared_flags (state))
+ != bool (aarch64_fntype_shared_flags (fntype, state)))
+ return false;
return true;
}
new file mode 100644
@@ -0,0 +1,51 @@
+// { dg-options "-O2 -fno-schedule-insns -fno-schedule-insns2" }
+// { dg-final { check-function-bodies "**" "" } }
+
+#pragma GCC target "+sme2"
+
+void gen_zt0() __arm_preserves("za") __arm_out("zt0");
+void callee() __arm_inout("za");
+
+/*
+** caller_inout:
+** ...
+** str zt0, \[[^\n]+\]
+** bl callee
+** ldr zt0, \[[^\n]+\]
+** ...
+** ret
+*/
+void caller_inout() __arm_inout("za", "zt0") { callee(); }
+
+/*
+** caller_in:
+** ...
+** str zt0, \[[^\n]+\]
+** bl callee
+** ldr zt0, \[[^\n]+\]
+** ...
+** ret
+*/
+void caller_in() __arm_inout("za") __arm_in("zt0") { callee(); }
+
+/*
+** caller_out:
+** ...
+** str zt0, \[[^\n]+\]
+** bl callee
+** ldr zt0, \[[^\n]+\]
+** ...
+** ret
+*/
+void caller_out() __arm_inout("za") __arm_in("zt0") { gen_zt0(); callee(); }
+
+/*
+** caller_preserves:
+** ...
+** str zt0, \[[^\n]+\]
+** bl callee
+** ldr zt0, \[[^\n]+\]
+** ...
+** ret
+*/
+void caller_preserves() __arm_inout("za") __arm_preserves("zt0") { callee(); }