92 machine_get_matrix_cpu_state(struct arm_matrix_cpu_state *cpu_state)  in machine_get_matrix_cpu_state()  argument
95 	cpu_state->have_sme = arm_sme_version() > 0;  in machine_get_matrix_cpu_state()
96 	if (cpu_state->have_sme) {  in machine_get_matrix_cpu_state()
97 		cpu_state->za_is_enabled = !!(__builtin_arm_rsr64("SVCR") & SVCR_ZA);  in machine_get_matrix_cpu_state()
99 		cpu_state->za_is_enabled = false;  in machine_get_matrix_cpu_state()
146 …text(thread_t old, arm_sme_saved_state_t *old_sme_ss, const struct arm_matrix_cpu_state *cpu_state)  in machine_save_sme_context()  argument
158 	if (cpu_state->za_is_enabled) {  in machine_save_sme_context()
164 …hread_t new, const arm_sme_saved_state_t *new_sme_ss, const struct arm_matrix_cpu_state *cpu_state)  in machine_restore_sme_context()  argument
170 			if (!cpu_state->za_is_enabled) {  in machine_restore_sme_context()
174 		} else if (cpu_state->za_is_enabled) {  in machine_restore_sme_context()
183 machine_disable_sme_context(const struct arm_matrix_cpu_state *cpu_state)  in machine_disable_sme_context()  argument
185 	if (cpu_state->za_is_enabled) {  in machine_disable_sme_context()
198 	struct arm_matrix_cpu_state cpu_state;  in machine_switch_matrix_context()  local
199 	machine_get_matrix_cpu_state(&cpu_state);  in machine_switch_matrix_context()
206 	if (cpu_state.have_sme) {  in machine_switch_matrix_context()
207 		machine_save_sme_context(old, old_sme_ss, &cpu_state);  in machine_switch_matrix_context()
213 	if (cpu_state.have_sme && !new_sme_ss) {  in machine_switch_matrix_context()
214 		machine_disable_sme_context(&cpu_state);  in machine_switch_matrix_context()
220 	if (cpu_state.have_sme) {  in machine_switch_matrix_context()
221 		machine_restore_sme_context(new, new_sme_ss, &cpu_state);  in machine_switch_matrix_context()