diff options
Diffstat (limited to 'src/core/NEON/wrapper/svtraits.h')
-rw-r--r-- | src/core/NEON/wrapper/svtraits.h | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/src/core/NEON/wrapper/svtraits.h b/src/core/NEON/wrapper/svtraits.h index 465983d16f..330d272752 100644 --- a/src/core/NEON/wrapper/svtraits.h +++ b/src/core/NEON/wrapper/svtraits.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2021 Arm Limited. + * Copyright (c) 2021-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -23,8 +23,9 @@ */ #ifndef SRC_CORE_NEON_WRAPPER_SVTRAITS_H #define SRC_CORE_NEON_WRAPPER_SVTRAITS_H -#if defined(__ARM_FEATURE_SVE) +#if defined(ARM_COMPUTE_ENABLE_SVE) #include "src/core/NEON/SVEMath.h" + #include <arm_sve.h> namespace arm_compute @@ -59,12 +60,15 @@ DEFINE_TYPES(uint64_t) DEFINE_TYPES(float16_t) DEFINE_TYPES(float32_t) DEFINE_TYPES(float64_t) + +#if __ARM_FEATURE_SVE_BF16 DEFINE_TYPES(bfloat16_t) +#endif // #if __ARM_FEATURE_SVE_BF16 #undef DEFINE_TYPES } // namespace wrapper } // namespace arm_compute -#endif /* defined(__ARM_FEATURE_SVE) */ +#endif /* defined(ARM_COMPUTE_ENABLE_SVE) */ #endif /* #ifndef SRC_CORE_NEON_WRAPPER_SVTRAITS_H */ |