Remove warnings on non-buffer tensor constants (#148483)

tugsbayasgalan · pytorchmergebot · commit fb0e9cb0a0a1 · 2025-03-12T18:20:04.000Z
Export already registers tensor constants directly in the graph and this is also true for Torchbind objects. This removes warning that pollutes the output. Differential Revision: [D70577856](https://our.internmc.facebook.com/intern/diff/D70577856) Pull Request resolved: #148483 Approved by: https://github.com/zhxchen17, https://github.com/zou3519 ghstack dependencies: #148364
diff --git a/torch/fx/graph.py b/torch/fx/graph.py
@@ -1726,8 +1726,6 @@ def check_arg(arg: Node, n: Optional[Node] = None) -> None:
 
         # Check targets are legit
         if self.owning_module:
-            num_warnings = 0
-            MAX_WARNINGS = 5
             for node in self.nodes:
                 if node.op == "call_function":
                     if not callable(node.target):
@@ -1759,29 +1757,8 @@ def check_arg(arg: Node, n: Optional[Node] = None) -> None:
                                 f"Node {node} target {node.target} {atom} of {seen_qualname} does "
                                 "not reference an nn.Module"
                             )
-                        elif (
-                            node.op == "get_attr"
-                            and not isinstance(new_m_itr, torch.nn.Module)
-                            and not isinstance(new_m_itr, torch.nn.Parameter)
-                            and atom not in m_itr._buffers
-                        ):
-                            if num_warnings < MAX_WARNINGS:
-                                # Don't emit this warning too frequently,
-                                # for very large graphs this can become very expensive
-                                # from a performance perspective.
-                                warnings.warn(
-                                    f"Node {node} target {node.target} {atom} of {seen_qualname} does "
-                                    "not reference an nn.Module, nn.Parameter, or buffer, which is "
-                                    "what 'get_attr' Nodes typically target"
-                                )
-                            num_warnings += 1
-                        else:
-                            m_itr = new_m_itr
-            if num_warnings > MAX_WARNINGS:
-                warnings.warn(
-                    f"Additional {num_warnings - MAX_WARNINGS} warnings "
-                    "suppressed about get_attr references"
-                )
+
+                        m_itr = new_m_itr
 
     @compatibility(is_backward_compatible=True)
     def eliminate_dead_code(
