diff options
Diffstat (limited to 'support/Bfloat16.h')
-rw-r--r-- | support/Bfloat16.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/support/Bfloat16.h b/support/Bfloat16.h index 173f2d16e2..5fd45cf209 100644 --- a/support/Bfloat16.h +++ b/support/Bfloat16.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020-2021 Arm Limited. + * Copyright (c) 2020-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -40,7 +40,7 @@ namespace inline uint16_t float_to_bf16(const float v) { const uint32_t *fromptr = reinterpret_cast<const uint32_t *>(&v); -#if defined(__ARM_FEATURE_BF16_VECTOR_ARITHMETIC) || defined(ARM_COMPUTE_FORCE_BF16) +#if defined(ARM_COMPUTE_ENABLE_BF16) uint16_t res; __asm __volatile( @@ -50,7 +50,7 @@ inline uint16_t float_to_bf16(const float v) : : [fromptr] "r"(fromptr), [toptr] "r"(&res) : "v0", "memory"); -#else /* defined(__ARM_FEATURE_BF16_VECTOR_ARITHMETIC) || defined(ARM_COMPUTE_FORCE_BF16) */ +#else /* defined(ARM_COMPUTE_ENABLE_BF16) */ uint16_t res = (*fromptr >> 16); const uint16_t error = (*fromptr & 0x0000ffff); uint16_t bf_l = res & 0x0001; @@ -58,7 +58,7 @@ inline uint16_t float_to_bf16(const float v) { res += 1; } -#endif /* defined(__ARM_FEATURE_BF16_VECTOR_ARITHMETIC) || defined(ARM_COMPUTE_FORCE_BF16) */ +#endif /* defined(ARM_COMPUTE_ENABLE_BF16) */ return res; } |