Suppress more warnings

tugsbayasgalan · tugsbayasgalan · commit fbd1ded84a6b · 2025-03-05T08:58:14.000-08:00
ghstack-source-id: 7a8cf6c Pull Request resolved: #148488
diff --git a/torch/_export/passes/lift_constants_pass.py b/torch/_export/passes/lift_constants_pass.py
@@ -1,5 +1,6 @@
 # mypy: allow-untyped-defs
 import collections
+import logging
 import warnings
 from typing import Any, Union
 
@@ -18,6 +19,8 @@
 )
 from torch.fx.graph_module import _get_attr
 
+log = logging.getLogger(__name__)
+
 
 class ConstantAttrMap(collections.abc.MutableMapping):
     """A mapping class that understands how to use module constants (tensors,
@@ -198,7 +201,7 @@ def lift_constants_pass(
             elif isinstance(constant_val, torch.Tensor):
                 # Remove the parameterness of constant_val
                 if isinstance(constant_val, torch.nn.Parameter):
-                    warnings.warn(
+                    log.debug(
                         f"{node.target} created when tracing {node.meta.get('stack_trace', '<unknown stack>')} is a parameter. But"
                         f"it's not registered with register_parameter(). export will treat it as a constant tensor"
                     )
diff --git a/torch/export/_unlift.py b/torch/export/_unlift.py
@@ -78,7 +78,12 @@ def _unlift_inputs_as_getattr(
 
         else:
             with gm.graph.inserting_after(input_node):
-                getattr_node = gm.graph.get_attr(lifted_node)
+                # It is fine to ignore this warning because
+                # it is guaranteed that we will populate this 
+                # attr later. 
+                with warnings.catch_warnings():
+                    warnings.simplefilter("ignore")
+                    getattr_node = gm.graph.get_attr(lifted_node)
                 input_node.replace_all_uses_with(getattr_node)
                 metadata = input_node.meta
                 gm.graph.erase_node(input_node)
diff --git a/torch/fx/experimental/proxy_tensor.py b/torch/fx/experimental/proxy_tensor.py
@@ -14,7 +14,6 @@
 import traceback
 import typing
 import typing_extensions
-import warnings
 import weakref
 from collections import defaultdict
 from collections.abc import Generator, Mapping, Sequence
@@ -1799,7 +1798,7 @@ def call_module(
         try:
             return Tracer.call_module(self, m, forward, args, kwargs)
         except _ModuleNotInstalledAsSubmoduleError:
-            warnings.warn(
+            log.debug(
                 f"Unable to find the path of the module {m}. "
                 "This might be because the module was not properly registered "
                 "as a submodule, which is not good practice. We will trace "
