diff options
Diffstat (limited to 'src/core/common/Registrars.h')
-rw-r--r-- | src/core/common/Registrars.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/core/common/Registrars.h b/src/core/common/Registrars.h index cc76de2be5..42c1aaa9fa 100644 --- a/src/core/common/Registrars.h +++ b/src/core/common/Registrars.h @@ -167,10 +167,10 @@ #define REGISTER_INTEGER_SVE2(func_name) nullptr #endif /* defined(ENABLE_INTEGER_KERNELS) */ -#if defined(__ARM_FEATURE_BF16_VECTOR_ARITHMETIC) || defined(ARM_COMPUTE_FORCE_BF16) +#if defined(ARM_COMPUTE_ENABLE_BF16) #define REGISTER_BF16_NEON(func_name) &(func_name) -#else /* !(defined(__ARM_FEATURE_BF16_VECTOR_ARITHMETIC) || defined(ARM_COMPUTE_FORCE_BF16))*/ +#else /* !(defined(ARM_COMPUTE_ENABLE_BF16))*/ #define REGISTER_BF16_NEON(func_name) nullptr -#endif /* defined(__ARM_FEATURE_BF16_VECTOR_ARITHMETIC) || defined(ARM_COMPUTE_FORCE_BF16)*/ +#endif /* defined(ARM_COMPUTE_ENABLE_BF16)*/ #endif /* SRC_CORE_COMMON_REGISTRARS_H */ |