diff options
| author | David Beck <david.beck@arm.com> | 2018-10-22 13:16:00 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:54 +0100 |
| commit | 056be3cbe0306f4e1409c10952a8a73676a4acdd (patch) | |
| tree | c7bcbeed66468b49dde7cf16f77256cd128b584c /src/armnn/Runtime.cpp | |
| parent | a8e06ed540a934f966679e1ef1cf7acf295211b3 (diff) | |
| download | armnn-056be3cbe0306f4e1409c10952a8a73676a4acdd.tar.gz | |
IVGCVSW-1991 : refactor m_SupportedComputes in DeviceSpecs
Change-Id: Ied3d54dc356f5e4f87aeb59f66423ac1f893dd01
Diffstat (limited to 'src/armnn/Runtime.cpp')
| -rw-r--r-- | src/armnn/Runtime.cpp | 12 |
1 files changed, 3 insertions, 9 deletions
diff --git a/src/armnn/Runtime.cpp b/src/armnn/Runtime.cpp index 354a567f88..8a7023ed76 100644 --- a/src/armnn/Runtime.cpp +++ b/src/armnn/Runtime.cpp @@ -4,7 +4,8 @@ // #include "Runtime.hpp" -#include "armnn/Version.hpp" +#include <armnn/Version.hpp> +#include <backends/BackendRegistry.hpp> #include <iostream> @@ -133,16 +134,9 @@ Runtime::Runtime(const CreationOptions& options) : m_ClContextControl(options.m_GpuAccTunedParameters.get(), options.m_EnableGpuProfiling) , m_NetworkIdCounter(0) + , m_DeviceSpec{BackendRegistryInstance().GetBackendIds()} { BOOST_LOG_TRIVIAL(info) << "ArmNN v" << ARMNN_VERSION << "\n"; - - m_DeviceSpec.m_SupportedComputeDevices.insert(armnn::Compute::CpuRef); - #if ARMCOMPUTECL_ENABLED - m_DeviceSpec.m_SupportedComputeDevices.insert(armnn::Compute::GpuAcc); - #endif - #if ARMCOMPUTENEON_ENABLED - m_DeviceSpec.m_SupportedComputeDevices.insert(armnn::Compute::CpuAcc); - #endif } Runtime::~Runtime() |