diff options
author | Finn Williams <finn.williams@arm.com> | 2021-10-28 19:07:32 +0100 |
---|---|---|
committer | Finn Williams <finn.williams@arm.com> | 2021-11-08 14:33:17 +0000 |
commit | b1aad4270fa8ad5c4aa62e27d564baf723b2cee5 (patch) | |
tree | 98b19ba85b50e2c730d5d2e3822cd2b1438bd149 /src/backends/backendsCommon/common.mk | |
parent | 3f22d27f51c493e37b9da0692b6bf776f4430dcf (diff) | |
download | armnn-b1aad4270fa8ad5c4aa62e27d564baf723b2cee5.tar.gz |
IVGCVSW-6527 Support the new memory API in loaded network
* enable external memory management for neon and ref backends
* change m_TensorMemoryVector to hold shared pointers
* change input layer backend Id to match backend id of connected layer
Signed-off-by: Finn Williams <finn.williams@arm.com>
Change-Id: I2216a724028312eb101b290df3f224177826b1a0
Diffstat (limited to 'src/backends/backendsCommon/common.mk')
-rw-r--r-- | src/backends/backendsCommon/common.mk | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/common.mk b/src/backends/backendsCommon/common.mk index a77ec06035..56c9d6545a 100644 --- a/src/backends/backendsCommon/common.mk +++ b/src/backends/backendsCommon/common.mk @@ -17,6 +17,7 @@ COMMON_SOURCES := \ MapWorkload.cpp \ MemCopyWorkload.cpp \ MemImportWorkload.cpp \ + MemoryManager.cpp \ MemSyncWorkload.cpp \ OptimizationViews.cpp \ TensorHandleFactoryRegistry.cpp \ @@ -25,7 +26,8 @@ COMMON_SOURCES := \ WorkloadFactory.cpp \ WorkloadUtils.cpp \ memoryOptimizerStrategyLibrary/strategies/ConstantMemoryStrategy.cpp \ - memoryOptimizerStrategyLibrary/strategies/StrategyValidator.cpp \ + memoryOptimizerStrategyLibrary/strategies/SingleAxisPriorityList.cpp \ + memoryOptimizerStrategyLibrary/strategies/StrategyValidator.cpp # COMMON_TEST_SOURCES contains the list of files to be included @@ -104,7 +106,8 @@ COMMON_TEST_SOURCES := \ test/layerTests/TransposeConvolution2dTestImpl.cpp \ test/layerTests/UnidirectionalSequenceLstmTestImpl.cpp \ memoryOptimizerStrategyLibrary/test/ConstMemoryStrategyTests.cpp \ - memoryOptimizerStrategyLibrary/test/ValidatorStrategyTests.cpp + memoryOptimizerStrategyLibrary/test/ValidatorStrategyTests.cpp \ + memoryOptimizerStrategyLibrary/test/SingleAxisPriorityListTests.cpp ifeq ($(ARMNN_REF_ENABLED),1) COMMON_TEST_SOURCES += \ |