Suppress more warnings

tugsbayasgalan · tugsbayasgalan · commit dbd7db76bcc2 · 2025-03-04T13:25:41.000-08:00
ghstack-source-id: 8f58615 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,9 @@ def _unlift_inputs_as_getattr(
 
         else:
             with gm.graph.inserting_after(input_node):
-                getattr_node = gm.graph.get_attr(lifted_node)
+                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)
