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:
-rw-r--r--src/arm/mach/init.c5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/arm/mach/init.c b/src/arm/mach/init.c
index dbea578..ef3aace 100644
--- a/src/arm/mach/init.c
+++ b/src/arm/mach/init.c
@@ -366,6 +366,11 @@ void cpuinfo_arm_mach_init(void) {
cpuinfo_isa.dot = true;
}
+ const uint32_t has_FEAT_BF16 = get_sys_info_by_name("hw.optional.arm.FEAT_BF16");
+ if (has_FEAT_BF16 != 0) {
+ cpuinfo_isa.bf16 = true;
+ }
+
uint32_t num_clusters = 1;
for (uint32_t i = 0; i < mach_topology.cores; i++) {
cores[i] = (struct cpuinfo_core) {