diff --git a/mace/core/memory_optimizer.cc b/mace/core/memory_optimizer.cc index b781682f618e79149b99dad5002ac68031989362..03e79f1e979aa77fcda8c5bbfb7ffa81fef486d6 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 7c7bb86517a96f011955cfd3b98a4f3b0050f9cb..5f5dce9569a8dac4f1c2ac579fc18edb9d0c29ee 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; }