From db82fc7ca6e6bf261f51a6794b400a04eaf55bd0 Mon Sep 17 00:00:00 2001
From: Yinghai Lu <yinghai@fb.com>
Date: Tue, 12 Feb 2019 23:59:40 -0800
Subject: [PATCH] Add more debugging facilities to ONNXIFI transform (#17043)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17043

Add more debugging facilities for ONXNIFI transform.

Reviewed By: ipiszy

Differential Revision: D14019492

fbshipit-source-id: 8c258ccba2f8ce77db096031fc8a61e15bd8af93
---
 caffe2/opt/onnxifi_transformer.cc | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/caffe2/opt/onnxifi_transformer.cc b/caffe2/opt/onnxifi_transformer.cc
index 3079d26..0fdbeb5 100644
--- a/caffe2/opt/onnxifi_transformer.cc
+++ b/caffe2/opt/onnxifi_transformer.cc
@@ -574,8 +574,12 @@ NetDef OnnxifiTransformer::SubnetToOnnxifiOpViaC2(
 
   // Debugging stuff
   if (opts_.debug) {
-    WriteProtoToTextFile(onnxifi_net, "debug_onnxifi_net.pb_txt");
-    WriteProtoToTextFile(net_opt, "debug_optimized_net.pb_txt");
+    WriteProtoToTextFile(
+        onnxifi_net,
+        "debug_onnxifi_net_" + c10::to_string(onnxifi_op_id_) + ".pb_txt");
+    WriteProtoToTextFile(
+        net_opt,
+        "debug_optimized_net_" + c10::to_string(onnxifi_op_id_) + ".pb_txt");
   }
   return net_opt;
 }
@@ -728,7 +732,12 @@ CaffeMap<std::string, TensorShape> OnnxifiTransformer::SsaRewriteAndMapNames(
   // Make sure weights do not contain output of any op.
   for (const auto& op : pred_net->op()) {
     for (const auto& output : op.output()) {
-      CAFFE_ENFORCE_EQ(weights.count(output), 0);
+      CAFFE_ENFORCE_EQ(
+          weights.count(output),
+          0,
+          "Weight ",
+          output,
+          " shouldn't appear in the output");
     }
   }
   input_mapping_ = onnx::SsaRewrite(nullptr, pred_net, weights);
-- 
2.7.4