diff options
Diffstat (limited to 'src/cpu/kernels/cast/generic')
-rw-r--r-- | src/cpu/kernels/cast/generic/neon/bfloat16.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/cpu/kernels/cast/generic/neon/bfloat16.cpp b/src/cpu/kernels/cast/generic/neon/bfloat16.cpp index b15584b0aa..aac4ef4ca0 100644 --- a/src/cpu/kernels/cast/generic/neon/bfloat16.cpp +++ b/src/cpu/kernels/cast/generic/neon/bfloat16.cpp @@ -24,6 +24,7 @@ #if defined(__ARM_FEATURE_BF16_VECTOR_ARITHMETIC) || defined(ARM_COMPUTE_FORCE_BF16) #include "arm_compute/core/TensorInfo.h" +#include "src/core/NEON/wrapper/wrapper.h" #include "src/cpu/kernels/CpuCastKernel.h" #include "src/cpu/kernels/cast/list.h" #include "support/SaturateCast.h" |