diff options
Diffstat (limited to 'src/cpu/kernels/elementwise_binary/generic/sve/fp16.cpp')
| -rw-r--r-- | src/cpu/kernels/elementwise_binary/generic/sve/fp16.cpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/cpu/kernels/elementwise_binary/generic/sve/fp16.cpp b/src/cpu/kernels/elementwise_binary/generic/sve/fp16.cpp index d764f56623..8adacbfe67 100644 --- a/src/cpu/kernels/elementwise_binary/generic/sve/fp16.cpp +++ b/src/cpu/kernels/elementwise_binary/generic/sve/fp16.cpp @@ -21,7 +21,9 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ -#if defined(ARM_COMPUTE_ENABLE_SVE) + +#if defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS) + #include "arm_compute/core/Helpers.h" #include "src/cpu/kernels/elementwise_binary/generic/sve/impl.h" namespace arm_compute @@ -58,4 +60,5 @@ template void sve_fp16_comparison_elementwise_binary<ComparisonOperation::LessEq } // namespace cpu } // namespace arm_compute -#endif //defined(ARM_COMPUTE_ENABLE_SVE) + +#endif /* defined(__ARM_FEATURE_FP16_VECTOR_ARITHMETIC) && defined(ENABLE_FP16_KERNELS) */
\ No newline at end of file |