diff options
Diffstat (limited to 'src/backends/aclCommon')
| -rw-r--r-- | src/backends/aclCommon/BaseMemoryManager.cpp | 2 | ||||
| -rw-r--r-- | src/backends/aclCommon/BaseMemoryManager.hpp | 6 |
2 files changed, 2 insertions, 6 deletions
diff --git a/src/backends/aclCommon/BaseMemoryManager.cpp b/src/backends/aclCommon/BaseMemoryManager.cpp index 7c06ec5537..844fbcd4ca 100644 --- a/src/backends/aclCommon/BaseMemoryManager.cpp +++ b/src/backends/aclCommon/BaseMemoryManager.cpp @@ -101,7 +101,7 @@ NeonMemoryManager::CreateMemoryGroup(const std::shared_ptr<arm_compute::MemoryMa std::shared_ptr<arm_compute::IMemoryGroup> ClMemoryManager::CreateMemoryGroup(const std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager) { - return std::make_shared<arm_compute::CLMemoryGroup>(memoryManager); + return std::make_shared<arm_compute::MemoryGroup>(memoryManager); } #endif diff --git a/src/backends/aclCommon/BaseMemoryManager.hpp b/src/backends/aclCommon/BaseMemoryManager.hpp index b8d1922d62..adef8bc713 100644 --- a/src/backends/aclCommon/BaseMemoryManager.hpp +++ b/src/backends/aclCommon/BaseMemoryManager.hpp @@ -7,14 +7,10 @@ #include <backendsCommon/IMemoryManager.hpp> #include <backendsCommon/WorkloadFactory.hpp> -#if defined(ARMCOMPUTENEON_ENABLED) +#if defined(ARMCOMPUTENEON_ENABLED) || defined(ARMCOMPUTECL_ENABLED) #include <arm_compute/runtime/MemoryGroup.h> #endif -#if defined(ARMCOMPUTECL_ENABLED) -#include <arm_compute/runtime/CL/CLMemoryGroup.h> -#endif - #if defined(ARMCOMPUTENEON_ENABLED) || defined(ARMCOMPUTECL_ENABLED) #include <arm_compute/runtime/IAllocator.h> #include <arm_compute/runtime/IMemoryGroup.h> |