diff options
Diffstat (limited to 'src/core')
-rw-r--r-- | src/core/NEON/wrapper/intrinsics/svdup_n.h | 6 | ||||
-rw-r--r-- | src/core/NEON/wrapper/svtraits.h | 5 |
2 files changed, 8 insertions, 3 deletions
diff --git a/src/core/NEON/wrapper/intrinsics/svdup_n.h b/src/core/NEON/wrapper/intrinsics/svdup_n.h index b1aed97d9c..9c42c86db7 100644 --- a/src/core/NEON/wrapper/intrinsics/svdup_n.h +++ b/src/core/NEON/wrapper/intrinsics/svdup_n.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020 Arm Limited. + * Copyright (c) 2020, 2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -46,7 +46,9 @@ SVDUP_N_IMPL(uint64_t, svuint64_t, u64) SVDUP_N_IMPL(float16_t, svfloat16_t, f16) SVDUP_N_IMPL(float, svfloat32_t, f32) SVDUP_N_IMPL(float64_t, svfloat64_t, f64) +#if __ARM_FEATURE_SVE_BF16 SVDUP_N_IMPL(bfloat16_t, svbfloat16_t, bf16) +#endif // #if __ARM_FEATURE_SVE_BF16 #undef SVDUP_N_IMPL @@ -54,4 +56,4 @@ SVDUP_N_IMPL(bfloat16_t, svbfloat16_t, bf16) } // namespace arm_compute #endif /* defined(__ARM_FEATURE_SVE) */ -#endif /* SRC_CORE_NEON_WRAPPER_INTRINSICS_SVDUP_N_H */
\ No newline at end of file +#endif /* SRC_CORE_NEON_WRAPPER_INTRINSICS_SVDUP_N_H */ diff --git a/src/core/NEON/wrapper/svtraits.h b/src/core/NEON/wrapper/svtraits.h index 1d599a246c..5ccd0ba8f1 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 * @@ -59,7 +59,10 @@ 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 |