From fe6af7cb8dc6d49320352dfa66e92f06d4289662 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Sat, 9 Jan 2021 13:46:42 +0800
Subject: [PATCH] fix(mgb): fix fastrun log

GitOrigin-RevId: 0a6cd47e492f3b88987d4195351735332fb5ecca
---
 src/opr/impl/search_policy/algo_chooser.cpp | 33 +++++++++++++++++++++--------
 1 file changed, 24 insertions(+), 9 deletions(-)

diff --git a/src/opr/impl/search_policy/algo_chooser.cpp b/src/opr/impl/search_policy/algo_chooser.cpp
index 74d619c1..b6e07a2c 100644
--- a/src/opr/impl/search_policy/algo_chooser.cpp
+++ b/src/opr/impl/search_policy/algo_chooser.cpp
@@ -185,15 +185,30 @@ size_t AlgoChooser<Opr>::setup_algo(const TensorLayoutArray& layouts,
         algo = get_algo(ctx);
     }
     size_t workspace = ctx.get_workspace_size_bytes(algo);
-    mgb_log_debug(
-            "%s: tensor layouts(%s %s, %s %s) -> (%s %s): algo=%s "
-            "workspace=%.2fMiB reproducible=%d",
-            mgb_opr->dyn_typeinfo()->name, layouts[0].to_string().c_str(),
-            layouts[0].dtype.name(), layouts[1].to_string().c_str(),
-            layouts[1].dtype.name(),
-            layouts[layouts.size() - 1].to_string().c_str(),
-            layouts[layouts.size() - 1].dtype.name(), algo.name.c_str(),
-            workspace / (1024 * 1024.0), algo.is_reproducible);
+
+    std::string ret;
+    ret.append(mgb_opr->dyn_typeinfo()->name);
+    ret.append(": tensor layouts(");
+    for (size_t i = 0; i < arity_in; ++i) {
+        if (i) {
+            ret.append(", ");
+        }
+        ret.append(layouts[i].to_string() + " ");
+        ret.append(layouts[i].dtype.name());
+    }
+    ret.append(") -> (");
+    for (size_t i = 0; i < arity_out; ++i) {
+        if (i) {
+            ret.append(", ");
+        }
+        ret.append(layouts[i + arity_in].to_string() + " ");
+        ret.append(layouts[i + arity_in].dtype.name());
+    }
+    ret.append("): algo=" + algo.name);
+    ret.append(ssprintf(" workspace=%.2fMiB reproducible=%d",
+                        workspace / (1024 * 1024.0), algo.is_reproducible));
+    mgb_log_debug("%s", ret.c_str());
+
     megdnn_opr->execution_policy() = {algo};
     return workspace;
 }