diff options
Diffstat (limited to 'src/cpu/CpuContext.h')
| -rw-r--r-- | src/cpu/CpuContext.h | 14 |
1 files changed, 3 insertions, 11 deletions
diff --git a/src/cpu/CpuContext.h b/src/cpu/CpuContext.h index e909767a7b..9a59af39c1 100644 --- a/src/cpu/CpuContext.h +++ b/src/cpu/CpuContext.h @@ -26,6 +26,7 @@ #include "src/common/AllocatorWrapper.h" #include "src/common/IContext.h" +#include "src/common/cpuinfo/CpuInfo.h" namespace arm_compute { @@ -34,17 +35,8 @@ namespace cpu /** Structure that encodes the CPU capabilities to be used */ struct CpuCapabilities { - bool neon{ false }; - bool sve{ false }; - bool sve2{ false }; - - bool fp16{ false }; - bool bf16{ false }; - bool dot{ false }; - bool mmla_int8{ false }; - bool mmla_fp{ false }; - - int32_t max_threads{ -1 }; + cpuinfo::CpuInfo cpu_info{}; + int32_t max_threads{ -1 }; }; /** CPU context implementation class */ |