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

Signed-off-by: Naren Dasan <naren@narendasan.com> Signed-off-by: Naren Dasan <narens@nvidia.com>
pytorch · Jan 23, 2021 · 86bb5b7 · 86bb5b7
1 parent 6b942e5
commit 86bb5b7
Showing 1 changed file with 1 addition and 1 deletion.
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: