diff options
Diffstat (limited to 'src/cpu/kernels')
-rw-r--r-- | src/cpu/kernels/cast/generic/neon/bfloat16.cpp | 1 | ||||
-rw-r--r-- | src/cpu/kernels/cast/generic/neon/fp16.cpp | 1 |
2 files changed, 2 insertions, 0 deletions
diff --git a/src/cpu/kernels/cast/generic/neon/bfloat16.cpp b/src/cpu/kernels/cast/generic/neon/bfloat16.cpp index 91c15be279..d8e2756192 100644 --- a/src/cpu/kernels/cast/generic/neon/bfloat16.cpp +++ b/src/cpu/kernels/cast/generic/neon/bfloat16.cpp @@ -23,6 +23,7 @@ */ #if defined(ARM_COMPUTE_ENABLE_BF16) +#include "arm_compute/core/CPP/CPPTypes.h" #include "arm_compute/core/TensorInfo.h" #include "src/core/NEON/wrapper/wrapper.h" #include "src/cpu/kernels/CpuCastKernel.h" diff --git a/src/cpu/kernels/cast/generic/neon/fp16.cpp b/src/cpu/kernels/cast/generic/neon/fp16.cpp index 385ca1898d..6cd0c8500b 100644 --- a/src/cpu/kernels/cast/generic/neon/fp16.cpp +++ b/src/cpu/kernels/cast/generic/neon/fp16.cpp @@ -23,6 +23,7 @@ */ #if defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS) +#include "arm_compute/core/CPP/CPPTypes.h" #include "arm_compute/core/TensorInfo.h" #include "src/cpu/kernels/CpuCastKernel.h" #include "src/cpu/kernels/cast/list.h" |