diff options
Diffstat (limited to 'src/backends/neon/NeonBackend.hpp')
-rw-r--r-- | src/backends/neon/NeonBackend.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/backends/neon/NeonBackend.hpp b/src/backends/neon/NeonBackend.hpp index 3364e4be41..ef5258b657 100644 --- a/src/backends/neon/NeonBackend.hpp +++ b/src/backends/neon/NeonBackend.hpp @@ -6,6 +6,8 @@ #include <armnn/backends/IBackendInternal.hpp> +#include <arm_compute/core/CPP/CPPTypes.h> + namespace armnn { @@ -19,7 +21,8 @@ const BackendCapabilities cpuAccCapabilities("CpuAcc", {"PreImportIOTensors", false}, {"ExternallyManagedMemory", true}, {"MultiAxisPacking", false}, - {"SingleAxisPacking", true} + {"SingleAxisPacking", true}, + {"HasFp16", arm_compute::CPUInfo::get().has_fp16()} }); |