diff options
Diffstat (limited to 'src/backends/cl/ClTensorHandleFactory.cpp')
-rw-r--r-- | src/backends/cl/ClTensorHandleFactory.cpp | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/src/backends/cl/ClTensorHandleFactory.cpp b/src/backends/cl/ClTensorHandleFactory.cpp index 33995f7b34..237f27a4ed 100644 --- a/src/backends/cl/ClTensorHandleFactory.cpp +++ b/src/backends/cl/ClTensorHandleFactory.cpp @@ -73,10 +73,11 @@ std::unique_ptr<ITensorHandle> ClTensorHandleFactory::CreateTensorHandle(const T const bool IsMemoryManaged) const { std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo); - if (IsMemoryManaged) + if (!IsMemoryManaged) { - tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup()); + ARMNN_LOG(warning) << "ClTensorHandleFactory only has support for memory managed."; } + tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup()); return tensorHandle; } @@ -85,10 +86,11 @@ std::unique_ptr<ITensorHandle> ClTensorHandleFactory::CreateTensorHandle(const T const bool IsMemoryManaged) const { std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo, dataLayout); - if (IsMemoryManaged) + if (!IsMemoryManaged) { - tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup()); + ARMNN_LOG(warning) << "ClTensorHandleFactory only has support for memory managed."; } + tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup()); return tensorHandle; } |