87 arm_get_matrix_cpu_state(struct arm_matrix_cpu_state *cpu_state)  in arm_get_matrix_cpu_state()  argument
90 	cpu_state->have_sme = arm_sme_version() > 0;  in arm_get_matrix_cpu_state()
91 	if (cpu_state->have_sme) {  in arm_get_matrix_cpu_state()
92 		cpu_state->za_is_enabled = !!(__builtin_arm_rsr64("SVCR") & SVCR_ZA);  in arm_get_matrix_cpu_state()
94 		cpu_state->za_is_enabled = false;  in arm_get_matrix_cpu_state()
99 #pragma unused(cpu_state)  in arm_get_matrix_cpu_state()
144 …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
156 	if (cpu_state->za_is_enabled) {  in machine_save_sme_context()
162 …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
168 			if (!cpu_state->za_is_enabled) {  in machine_restore_sme_context()
172 		} else if (cpu_state->za_is_enabled) {  in machine_restore_sme_context()
179 machine_clear_sme_context(const struct arm_matrix_cpu_state *cpu_state)  in machine_clear_sme_context()  argument
181 	if (cpu_state->za_is_enabled) {  in machine_clear_sme_context()
191     const struct arm_matrix_cpu_state *cpu_state)  in machine_trap_el0_matrix_instructions()  argument
193 #pragma unused(cpu_state)  in machine_trap_el0_matrix_instructions()
205 	struct arm_matrix_cpu_state cpu_state;  in machine_switch_matrix_context()  local
206 	arm_get_matrix_cpu_state(&cpu_state);  in machine_switch_matrix_context()
214 	if (cpu_state.have_sme) {  in machine_switch_matrix_context()
215 		machine_save_sme_context(old, old_sme_ss, &cpu_state);  in machine_switch_matrix_context()
221 	if (cpu_state.have_sme && !new_sme_ss) {  in machine_switch_matrix_context()
222 		machine_clear_sme_context(&cpu_state);  in machine_switch_matrix_context()
228 	if (cpu_state.have_sme) {  in machine_switch_matrix_context()
229 		machine_restore_sme_context(new, new_sme_ss, &cpu_state);  in machine_switch_matrix_context()
237 	machine_trap_el0_matrix_instructions(enable_el0_trap, &cpu_state);  in machine_switch_matrix_context()