diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2021-09-10 11:40:54 +0100 |
|---|---|---|
| committer | finn.williams <finn.williams@arm.com> | 2021-09-14 11:33:33 +0000 |
| commit | c840263cb524bec4a1911b81d6300a34fe05a599 (patch) | |
| tree | 5d93f4c0b5295da85071c0381b29e3dd7171be70 | |
| parent | 284b5d19c8cf446b7eec16ea560c544ac39008c1 (diff) | |
| download | armnn-c840263cb524bec4a1911b81d6300a34fe05a599.tar.gz | |
IVGCVSW-6339 'IMemoryOptimizerStrategy Add backend capabilities'
* Added ExternallyManagedMemory capability, set false for backends
* Added MultiAxisPacking capability, set false for backends
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I96a2b66ed069a82552dae33f6e8ebe067bbd6c44
| -rw-r--r-- | src/backends/backendsCommon/test/CompatibilityTests.cpp | 12 | ||||
| -rw-r--r-- | src/backends/cl/ClBackend.hpp | 4 | ||||
| -rw-r--r-- | src/backends/neon/NeonBackend.hpp | 4 | ||||
| -rw-r--r-- | src/backends/reference/RefBackend.hpp | 4 |
4 files changed, 18 insertions, 6 deletions
diff --git a/src/backends/backendsCommon/test/CompatibilityTests.cpp b/src/backends/backendsCommon/test/CompatibilityTests.cpp index ff842e9c56..e1462e0d2e 100644 --- a/src/backends/backendsCommon/test/CompatibilityTests.cpp +++ b/src/backends/backendsCommon/test/CompatibilityTests.cpp @@ -180,7 +180,9 @@ TEST_CASE ("Ref_Backends_Capability_Test") {"AsyncExecution", true}, {"ProtectedContentAllocation", false}, {"ConstantTensorsAsInputs", true}, - {"PreImportIOTensors", false}}); + {"PreImportIOTensors", false}, + {"ExternallyManagedMemory", false}, + {"MultiAxisPacking", false}}); } #endif @@ -197,7 +199,9 @@ TEST_CASE ("Neon_Backends_Capability_Test") {"AsyncExecution", false}, {"ProtectedContentAllocation", false}, {"ConstantTensorsAsInputs", false}, - {"PreImportIOTensors", false}}); + {"PreImportIOTensors", false}, + {"ExternallyManagedMemory", false}, + {"MultiAxisPacking", false}}); } #endif @@ -214,7 +218,9 @@ TEST_CASE ("Cl_Backends_Capability_Test") {"AsyncExecution", false}, {"ProtectedContentAllocation", true}, {"ConstantTensorsAsInputs", false}, - {"PreImportIOTensors", false}}); + {"PreImportIOTensors", false}, + {"ExternallyManagedMemory", false}, + {"MultiAxisPacking", false}}); } #endif diff --git a/src/backends/cl/ClBackend.hpp b/src/backends/cl/ClBackend.hpp index 570d301616..0a069b930b 100644 --- a/src/backends/cl/ClBackend.hpp +++ b/src/backends/cl/ClBackend.hpp @@ -28,7 +28,9 @@ const BackendCapabilities gpuAccCapabilities("GpuAcc", {"AsyncExecution", false}, {"ProtectedContentAllocation", true}, {"ConstantTensorsAsInputs", false}, - {"PreImportIOTensors", false} + {"PreImportIOTensors", false}, + {"ExternallyManagedMemory", false}, + {"MultiAxisPacking", false} }); class ClBackend : public IBackendInternal diff --git a/src/backends/neon/NeonBackend.hpp b/src/backends/neon/NeonBackend.hpp index 0ae28d8a71..20da73aaf8 100644 --- a/src/backends/neon/NeonBackend.hpp +++ b/src/backends/neon/NeonBackend.hpp @@ -16,7 +16,9 @@ const BackendCapabilities cpuAccCapabilities("GpuAcc", {"AsyncExecution", false}, {"ProtectedContentAllocation", false}, {"ConstantTensorsAsInputs", false}, - {"PreImportIOTensors", false} + {"PreImportIOTensors", false}, + {"ExternallyManagedMemory", false}, + {"MultiAxisPacking", false} }); diff --git a/src/backends/reference/RefBackend.hpp b/src/backends/reference/RefBackend.hpp index 1a79d7090a..28c1591179 100644 --- a/src/backends/reference/RefBackend.hpp +++ b/src/backends/reference/RefBackend.hpp @@ -15,7 +15,9 @@ const BackendCapabilities cpuRefCapabilities("CpuRef", {"AsyncExecution", true}, {"ProtectedContentAllocation", false}, {"ConstantTensorsAsInputs", true}, - {"PreImportIOTensors", false} + {"PreImportIOTensors", false}, + {"ExternallyManagedMemory", false}, + {"MultiAxisPacking", false} }); const std::set<armnn::BackendCapability> oldCpuRefCapabilities { |