From 86bb5b7e351a25f043ffc0831ac44787fe89ea30 Mon Sep 17 00:00:00 2001
From: Naren Dasan <naren@narendasan.com>
Date: Fri, 22 Jan 2021 16:49:30 -0800
Subject: [PATCH] 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>
---
 core/lowering/passes/remove_bn_dim_check.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/core/lowering/passes/remove_bn_dim_check.cpp b/core/lowering/passes/remove_bn_dim_check.cpp
index 6116522484..9b2a63b0a7 100644
--- 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: