fix(//core/lowering): fix debug message for bn dim check removal pass

narendasan · narendasan · commit 86bb5b7e351a · 2021-01-22T16:49:30.000-08:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
diff --git a/core/lowering/passes/remove_bn_dim_check.cpp b/core/lowering/passes/remove_bn_dim_check.cpp
@@ -22,7 +22,7 @@ struct BNDimCheckRemoval {
   void run() {
     findBNDimCheckNodes(graph_->block());
     torch::jit::EliminateDeadCode(graph_);
-    LOG_GRAPH("Post aten::addmm branch fusion: " << *graph_);
+    LOG_GRAPH("Post batch norm dim check removal: " << *graph_);
   }
 
  private:

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ struct BNDimCheckRemoval {`
`22`	`22`	`void run() {`
`23`	`23`	`findBNDimCheckNodes(graph_->block());`
`24`	`24`	`torch::jit::EliminateDeadCode(graph_);`
`25`		`- LOG_GRAPH("Post aten::addmm branch fusion: " << *graph_);`
	`25`	`+ LOG_GRAPH("Post batch norm dim check removal: " << *graph_);`
`26`	`26`	`}`
`27`	`27`
`28`	`28`	`private:`