1 /* 2 * Copyright (C) 2022 Tino Reichardt <milky-zfs@mcmilk.de> 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 * 26 * $FreeBSD$ 27 */ 28 29 /* 30 * SIMD support: 31 * 32 * Following functions should be called to determine whether CPU feature 33 * is supported. All functions are usable in kernel and user space. 34 * If a SIMD algorithm is using more than one instruction set 35 * all relevant feature test functions should be called. 36 * 37 * Supported features: 38 * zfs_neon_available() 39 * zfs_sha256_available() 40 * zfs_sha512_available() 41 */ 42 43 #ifndef _FREEBSD_SIMD_AARCH64_H 44 #define _FREEBSD_SIMD_AARCH64_H 45 46 #include <sys/types.h> 47 #include <sys/ucontext.h> 48 #include <machine/elf.h> 49 #include <machine/fpu.h> 50 #include <machine/md_var.h> 51 #include <machine/pcb.h> 52 53 #ifdef _STANDALONE 54 55 #define kfpu_allowed() 0 56 #define kfpu_begin() do {} while (0) 57 #define kfpu_end() do {} while (0) 58 59 #else 60 61 #define kfpu_allowed() 1 62 #define kfpu_begin() do { \ 63 if (__predict_false(!is_fpu_kern_thread(0))) \ 64 fpu_kern_enter(curthread, NULL, FPU_KERN_NOCTX); \ 65 } while (0) 66 67 #define kfpu_end() do { \ 68 if (__predict_false(curthread->td_pcb->pcb_fpflags & PCB_FP_NOSAVE)) \ 69 fpu_kern_leave(curthread, NULL); \ 70 } while (0) 71 72 #endif 73 74 #define kfpu_initialize(tsk) do {} while (0) 75 #define kfpu_init() (0) 76 #define kfpu_fini() do {} while (0) 77 78 /* 79 * Check if NEON is available 80 */ 81 static inline boolean_t 82 zfs_neon_available(void) 83 { 84 return (elf_hwcap & HWCAP_FP); 85 } 86 87 /* 88 * Check if SHA256 is available 89 */ 90 static inline boolean_t 91 zfs_sha256_available(void) 92 { 93 return (elf_hwcap & HWCAP_SHA2); 94 } 95 96 /* 97 * Check if SHA512 is available 98 */ 99 static inline boolean_t 100 zfs_sha512_available(void) 101 { 102 return (elf_hwcap & HWCAP_SHA512); 103 } 104 105 #endif /* _FREEBSD_SIMD_AARCH64_H */ 106