diff options
Diffstat (limited to 'src/cpu/CpuContext.h')
| -rw-r--r-- | src/cpu/CpuContext.h | 24 |
1 files changed, 10 insertions, 14 deletions
diff --git a/src/cpu/CpuContext.h b/src/cpu/CpuContext.h index e909767a7b..0c8ae49f49 100644 --- a/src/cpu/CpuContext.h +++ b/src/cpu/CpuContext.h @@ -25,6 +25,7 @@ #define SRC_CPU_CPUCONTEXT_H #include "src/common/AllocatorWrapper.h" +#include "src/common/cpuinfo/CpuInfo.h" #include "src/common/IContext.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 */ @@ -68,8 +60,12 @@ public: AllocatorWrapper &allocator(); // Inherrited methods overridden - ITensorV2 *create_tensor(const AclTensorDescriptor &desc, bool allocate) override; - IQueue *create_queue(const AclQueueOptions *options) override; + ITensorV2 *create_tensor(const AclTensorDescriptor &desc, bool allocate) override; + IQueue *create_queue(const AclQueueOptions *options) override; + std::tuple<IOperator *, StatusCode> create_activation(const AclTensorDescriptor &src, + const AclTensorDescriptor &dst, + const AclActivationDescriptor &act, + bool is_validate) override; private: AllocatorWrapper _allocator; @@ -78,4 +74,4 @@ private: } // namespace cpu } // namespace arm_compute -#endif /* SRC_CPU_CPUCONTEXT_H */
\ No newline at end of file +#endif /* SRC_CPU_CPUCONTEXT_H */ |