Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/pytorch/cpuinfo.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKulin Seth <kulinseth@gmail.com>2022-07-15 19:25:34 +0300
committerGitHub <noreply@github.com>2022-07-15 19:25:34 +0300
commit60324e22653a37778a2c7e8c5281365f83e1ca81 (patch)
treea654fe8859cf51219e6cf6c6303a81c2b401bc5f /include
parent0c5d43135059ed9386782e12f4499038ca02e2e5 (diff)
Add support for Arm I8MM (#103)
Co-authored-by: Developer-Ecosystem-Engineering <65677710+Developer-Ecosystem-Engineering@users.noreply.github.com>
Diffstat (limited to 'include')
-rw-r--r--include/cpuinfo.h9
1 files changed, 9 insertions, 0 deletions
diff --git a/include/cpuinfo.h b/include/cpuinfo.h
index 7bcb931..e7ce9ed 100644
--- a/include/cpuinfo.h
+++ b/include/cpuinfo.h
@@ -1467,6 +1467,7 @@ static inline bool cpuinfo_has_x86_sha(void) {
bool bf16;
bool sve;
bool sve2;
+ bool i8mm;
#endif
bool rdm;
bool fp16arith;
@@ -1758,6 +1759,14 @@ static inline bool cpuinfo_has_arm_fcma(void) {
#endif
}
+static inline bool cpuinfo_has_arm_i8mm(void) {
+ #if CPUINFO_ARCH_ARM64
+ return cpuinfo_isa.i8mm;
+ #else
+ return false;
+ #endif
+}
+
static inline bool cpuinfo_has_arm_aes(void) {
#if CPUINFO_ARCH_ARM || CPUINFO_ARCH_ARM64
return cpuinfo_isa.aes;