Add logging for _MinimizerBase

Summary: Add logging so we know which nodes are currently being visited

Test Plan: lint & SC tests

Reviewed By: 842974287

Differential Revision: D30509865

fbshipit-source-id: 09e77e44c97c825242e0b24f90463b50f3ca19c6

diff --git a/torch/fx/passes/net_min_base.py b/torch/fx/passes/net_min_base.py
index 2a093be..b7a911e 100644 (file)
--- a/torch/fx/passes/net_min_base.py
+++ b/torch/fx/passes/net_min_base.py
@@ -1,5 +1,6 @@
 import argparse
 from typing import Any, Callable, Tuple, Dict, Optional
+import logging
 
 import torch
 import torch.fx
@@ -17,6 +18,8 @@ from .tools_common import (
     Names
 )
 
+_LOGGER = logging.getLogger(__name__)
+
 
 class FxNetMinimizerBadModuleError(Exception):
     """
@@ -403,6 +406,7 @@ class _MinimizerBase:
         culprits: NodeSet = set()
 
         for node in nodes:
+            _LOGGER.info(f"Visit node: {node.name}")
             cur_nodes: NodeSet = {node}
 
             if node in self.fusions: