From 4c3f716371da92977d4a998fe5c89dee04ddbced Mon Sep 17 00:00:00 2001 From: Viet-Hoa Do Date: Fri, 10 May 2024 15:21:01 +0100 Subject: Improve CPU extension detection on macos Resolves: COMPMID-7021 Signed-off-by: Viet-Hoa Do Change-Id: I809bc6ecd2845dfe6ee5de20a902aea4d07f15a5 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/11540 Benchmark: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Ramy Elgammal --- src/common/cpuinfo/CpuInfo.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/common/cpuinfo/CpuInfo.cpp b/src/common/cpuinfo/CpuInfo.cpp index 93f51e599a..809ab3e2c3 100644 --- a/src/common/cpuinfo/CpuInfo.cpp +++ b/src/common/cpuinfo/CpuInfo.cpp @@ -363,6 +363,8 @@ CpuInfo CpuInfo::build() isainfo.neon = get_hw_capability("hw.optional.neon"); isainfo.fp16 = get_hw_capability("hw.optional.neon_fp16"); isainfo.dot = get_hw_capability("hw.optional.arm.FEAT_DotProd"); + isainfo.bf16 = get_hw_capability("hw.optional.arm.FEAT_BF16"); + isainfo.i8mm = get_hw_capability("hw.optional.arm.FEAT_I8MM"); CpuInfo info(isainfo, cpus_model); return info; #elif defined(__aarch64__) && defined(_WIN64) /* #elif defined(__aarch64__) && defined(__APPLE__) */ -- cgit v1.2.1