aboutsummaryrefslogtreecommitdiff
path: root/tests/benchmark/NEON/SYSTEM/AlexNet.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/benchmark/NEON/SYSTEM/AlexNet.cpp')
-rw-r--r--tests/benchmark/NEON/SYSTEM/AlexNet.cpp6
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/benchmark/NEON/SYSTEM/AlexNet.cpp b/tests/benchmark/NEON/SYSTEM/AlexNet.cpp
index 1b90fbddfb..ad16d47b46 100644
--- a/tests/benchmark/NEON/SYSTEM/AlexNet.cpp
+++ b/tests/benchmark/NEON/SYSTEM/AlexNet.cpp
@@ -45,11 +45,11 @@ namespace test
{
namespace
{
-#ifdef ARM_COMPUTE_AARCH64_V8_2
+#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
const auto alex_net_data_types = framework::dataset::make("DataType", { DataType::F16, DataType::F32, DataType::QS8 });
-#else /* ARM_COMPUTE_AARCH64_V8_2 */
+#else /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */
const auto alex_net_data_types = framework::dataset::make("DataType", { DataType::F32, DataType::QS8 });
-#endif /* ARM_COMPUTE_AARCH64_V8_2 */
+#endif /* __ARM_FEATURE_FP16_VECTOR_ARITHMETIC */
} // namespace
using NEAlexNetFixture = AlexNetFixture<ITensor,