diff options
Diffstat (limited to 'src/armnn/BackendRegistry.cpp')
| -rw-r--r-- | src/armnn/BackendRegistry.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/armnn/BackendRegistry.cpp b/src/armnn/BackendRegistry.cpp index 7b1f6bcb5d..01f632fb22 100644 --- a/src/armnn/BackendRegistry.cpp +++ b/src/armnn/BackendRegistry.cpp @@ -5,7 +5,8 @@ #include <armnn/BackendRegistry.hpp> #include <armnn/Exceptions.hpp> -#include <ProfilingService.hpp> +#include <armnn/profiling/ArmNNProfiling.hpp> +#include <IProfilingService.hpp> namespace armnn { @@ -102,7 +103,7 @@ void BackendRegistry::Swap(BackendRegistry& instance, BackendRegistry::FactorySt std::swap(instance.m_Factories, other); } -void BackendRegistry::SetProfilingService(armnn::Optional<arm::pipe::ProfilingService&> profilingService) +void BackendRegistry::SetProfilingService(armnn::Optional<arm::pipe::IProfilingService&> profilingService) { m_ProfilingService = profilingService; } |