apple: 新增amx指令检测支持

nihui · Feb 23, 2024 · d34e4a9 · d34e4a9
1 parent 4b5d00b
commit d34e4a9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git 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
@@ -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)
@@ -292,6 +293,7 @@ RUAPU_ISAENTRY(sha3)
 RUAPU_ISAENTRY(sha512)
 RUAPU_ISAENTRY(sm3)
 RUAPU_ISAENTRY(sm4)
+RUAPU_ISAENTRY(amx)
 
 #elif __arm__ || defined(_M_ARM)
 RUAPU_ISAENTRY(edsp)