From 71141ac61c9864688db8934ac637986f31d5bd95 Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Fri, 26 Apr 2019 17:53:16 +0800
Subject: [PATCH] Add more log for locating out of range error of OpenCL.

---
 mace/core/memory_optimizer.cc | 11 +++++++++++
 mace/core/net_def_adapter.cc  |  2 +-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/mace/core/memory_optimizer.cc b/mace/core/memory_optimizer.cc
index b781682f..03e79f1e 100644
--- a/mace/core/memory_optimizer.cc
+++ b/mace/core/memory_optimizer.cc
@@ -275,6 +275,17 @@ std::string MemoryOptimizer::DebugInfo() const {
     sstream << "\n";
   }
 
+  for (auto tuple : tensor_mem_map_) {
+    sstream << tuple.first << ": " << tuple.second.mem_id << ", ";
+    if (mem_blocks_[tuple.second.mem_id].mem_type() == MemoryType::GPU_IMAGE) {
+      sstream << "[" << mem_blocks_[tuple.second.mem_id].x() << ", "
+              << mem_blocks_[tuple.second.mem_id].y() << "]";
+    } else {
+      sstream << "[" << mem_blocks_[tuple.second.mem_id].x() << "]";
+    }
+    sstream << "\n";
+  }
+
   return sstream.str();
 }
 
diff --git a/mace/core/net_def_adapter.cc b/mace/core/net_def_adapter.cc
index 7c7bb865..5f5dce95 100644
--- a/mace/core/net_def_adapter.cc
+++ b/mace/core/net_def_adapter.cc
@@ -294,7 +294,7 @@ MaceStatus NetDefAdapter::AdaptNetDef(
   }
 #endif  // MACE_ENABLE_OPENCL
 
-  VLOG(1) << DebugString(target_net_def);
+  VLOG(3) << DebugString(target_net_def);
   return MaceStatus::MACE_SUCCESS;
 }
 
-- 
GitLab