1 /* 2 * Copyright (c) 2022 Apple Computer, Inc. All rights reserved. 3 * 4 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@ 5 * 6 * This file contains Original Code and/or Modifications of Original Code 7 * as defined in and that are subject to the Apple Public Source License 8 * Version 2.0 (the 'License'). You may not use this file except in 9 * compliance with the License. The rights granted to you under the License 10 * may not be used to create, or enable the creation or redistribution of, 11 * unlawful or unlicensed copies of an Apple operating system, or to 12 * circumvent, violate, or enable the circumvention or violation of, any 13 * terms of an Apple operating system software license agreement. 14 * 15 * Please obtain a copy of the License at 16 * http://www.opensource.apple.com/apsl/ and read it before using this file. 17 * 18 * The Original Code and all software distributed under the License are 19 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, 21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, 22 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 23 * Please see the License for the specific language governing rights and 24 * limitations under the License. 25 * 26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@ 27 */ 28 29 #ifndef __ARM_MATRIX_H 30 #define __ARM_MATRIX_H 31 32 #include <mach/mach_types.h> 33 #include <mach/thread_status.h> 34 #include <stdbool.h> 35 #include <stddef.h> 36 #include <stdlib.h> 37 38 struct arm_matrix_operations { 39 const char *name; 40 41 size_t (*data_size)(void); 42 void *(*alloc_data)(void); 43 44 bool (*is_available)(void); 45 void (*start)(void); 46 void (*stop)(void); 47 48 void (*load_one_vector)(const void *); 49 void (*load_data)(const void *); 50 void (*store_data)(void *); 51 52 kern_return_t (*thread_get_state)(thread_act_t, void *); 53 kern_return_t (*thread_set_state)(thread_act_t, const void *); 54 }; 55 56 /** Operates on all SME context data. Legacy SIMD instructions between start() and stop() will trap. */ 57 extern const struct arm_matrix_operations sme_operations; 58 /** Operates only on SME context data controlled by `SVCR.ZA`. No effect on legacy SIMD instructions. */ 59 extern const struct arm_matrix_operations sme_za_operations; 60 61 #endif /* __ARM_MATRIX_H */ 62