From c840263cb524bec4a1911b81d6300a34fe05a599 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Fri, 10 Sep 2021 11:40:54 +0100 Subject: 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 Change-Id: I96a2b66ed069a82552dae33f6e8ebe067bbd6c44 --- src/backends/backendsCommon/test/CompatibilityTests.cpp | 12 +++++++++--- src/backends/cl/ClBackend.hpp | 4 +++- src/backends/neon/NeonBackend.hpp | 4 +++- 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 oldCpuRefCapabilities { -- cgit v1.2.1