2 files changed, 6 insertions, 1 deletions
diff --git a/src/graph2/backends/CL/CLDeviceBackend.cpp b/src/graph2/backends/CL/CLDeviceBackend.cpp
index 6d2d4f9b1a..71566d2f1f 100644
--- a/src/graph2/backends/CL/CLDeviceBackend.cpp
+++ b/src/graph2/backends/CL/CLDeviceBackend.cpp
@@ -101,7 +101,7 @@ void CLDeviceBackend::initialize_backend()
 void CLDeviceBackend::setup_backend_context(GraphContext &ctx)
 {
     // Setup tuner
-    set_kernel_tuning(ctx.is_tuning_enabled());
+    set_kernel_tuning(ctx.config().use_tuner);
 
     // Setup a management backend
     if(ctx.memory_management_ctx(Target::CL) == nullptr)
diff --git a/src/graph2/backends/NEON/NEDeviceBackend.cpp b/src/graph2/backends/NEON/NEDeviceBackend.cpp
index 9010c5d802..6cb507b4f1 100644
--- a/src/graph2/backends/NEON/NEDeviceBackend.cpp
+++ b/src/graph2/backends/NEON/NEDeviceBackend.cpp
@@ -40,6 +40,7 @@
 #include "arm_compute/runtime/MemoryManagerOnDemand.h"
 #include "arm_compute/runtime/OffsetLifetimeManager.h"
 #include "arm_compute/runtime/PoolManager.h"
+#include "arm_compute/runtime/Scheduler.h"
 
 #include "support/ToolchainSupport.h"
 
@@ -63,6 +64,10 @@ void NEDeviceBackend::initialize_backend()
 
 void NEDeviceBackend::setup_backend_context(GraphContext &ctx)
 {
+    // Set number of threads
+    Scheduler::get().set_num_threads(ctx.config().num_threads);
+
+    // Create function level memory manager
     if(ctx.memory_management_ctx(Target::NEON) == nullptr)
     {
         MemoryManagerContext mm_ctx;