From 1831bf381ec967198a4bc6599e692a0b6401a713 Mon Sep 17 00:00:00 2001 From: Monkeyking Date: Fri, 23 Feb 2024 13:04:27 +0800 Subject: [PATCH] apple: amx support (#28) --- README.md | 2 +- main.c | 1 + ruapu.h | 2 ++ 3 files changed, 4 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index d24d28a..17473b4 100644 --- a/README.md +++ b/README.md @@ -219,7 +219,7 @@ _`fma4` on zen1, ISA in hypervisor, etc._ |:---:|---| |x86|`mmx` `sse` `sse2` `sse3` `ssse3` `sse41` `sse42` `sse4a` `xop` `avx` `f16c` `fma` `fma4` `avx2` `avx512f` `avx512bw` `avx512cd` `avx512dq` `avx512vl` `avx512vnni` `avx512bf16` `avx512ifma` `avx512vbmi` `avx512vbmi2` `avx512fp16` `avxvnni` `avxvnniint8` `avxifma`| |arm|`edsp` `neon` `vfpv4`| -|aarch64|`neon` `vfpv4` `cpuid` `asimdhp` `asimddp` `asimdfhm` `bf16` `i8mm` `sve` `sve2` `svebf16` `svei8mm` `svef32mm` `sha3` `sha512` `sm3` `sm4`| +|aarch64|`neon` `vfpv4` `cpuid` `asimdhp` `asimddp` `asimdfhm` `bf16` `i8mm` `sve` `sve2` `svebf16` `svei8mm` `svef32mm` `sha3` `sha512` `sm3` `sm4` `amx`| |mips|`msa`| |powerpc|`vsx`| |s390x|`zvector`| diff --git a/main.c b/main.c index 8bb7afc..c8c4fc8 100644 --- a/main.c +++ b/main.c @@ -63,6 +63,7 @@ int main() PRINT_ISA_SUPPORT(sha512) PRINT_ISA_SUPPORT(sm3) PRINT_ISA_SUPPORT(sm4) + PRINT_ISA_SUPPORT(amx) #elif __arm__ || defined(_M_ARM) PRINT_ISA_SUPPORT(edsp) diff --git a/ruapu.h b/ruapu.h index 7d92933..14a495c 100644 --- a/ruapu.h +++ b/ruapu.h @@ -209,6 +209,7 @@ RUAPU_INSTCODE(sha3, 0xce000000) // eor3 v0.16b, v0.16b, v0.16b, v0.16b RUAPU_INSTCODE(sha512, 0xce608000) // sha512h q0, q0, v0.2d RUAPU_INSTCODE(sm3, 0xce60c000) // sm3partw1 v0.4s, v0.4s, v0.4s RUAPU_INSTCODE(sm4, 0xcec08400) // sm4e v0.4s, v0.4s +RUAPU_INSTCODE(amx, 0x00201220) // amx setup #elif __arm__ || defined(_M_ARM) @@ -295,6 +296,7 @@ RUAPU_ISAENTRY(sha3) RUAPU_ISAENTRY(sha512) RUAPU_ISAENTRY(sm3) RUAPU_ISAENTRY(sm4) +RUAPU_ISAENTRY(amx) #elif __arm__ || defined(_M_ARM) RUAPU_ISAENTRY(edsp)