aboutsummaryrefslogtreecommitdiff
path: root/tests/benchmark_new/main.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/benchmark_new/main.cpp')
-rw-r--r--tests/benchmark_new/main.cpp8
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/benchmark_new/main.cpp b/tests/benchmark_new/main.cpp
index 4c6811e372..46c51b6cfd 100644
--- a/tests/benchmark_new/main.cpp
+++ b/tests/benchmark_new/main.cpp
@@ -30,9 +30,9 @@
#include "support/ToolchainSupport.h"
#include "tests/TensorLibrary.h"
-#ifdef OPENCL
+#ifdef ARM_COMPUTE_CL
#include "arm_compute/runtime/CL/CLScheduler.h"
-#endif /* OPENCL */
+#endif /* ARM_COMPUTE_CL */
#include "arm_compute/runtime/Scheduler.h"
#include <fstream>
@@ -55,9 +55,9 @@ std::unique_ptr<TensorLibrary> library;
int main(int argc, char **argv)
{
-#ifdef OPENCL
+#ifdef ARM_COMPUTE_CL
CLScheduler::get().default_init();
-#endif /* OPENCL */
+#endif /* ARM_COMPUTE_CL */
framework::Framework &framework = framework::Framework::get();