always partition used static attr

Summary: For supported_operators that use static data, the data should always be partitioned along with that operator. This is required for adding addmm into supported_operator set because it allows us to partition in the weight and bias data.

Differential Revision: D49129705

fbshipit-source-id: dfd200221e42408128ec429c16672c2919a2ba72

Author: mcr229

backends/xnnpack/partition/xnnpack_partitioner.py

@@ -109,6 +109,11 @@ def check_constraint(node, ep) -> bool:
         return _OP_SUPPORT_CONSTRAINTS.get(node.target, lambda node, ep: True)(node, ep)
 
     def is_node_supported(self, submodules, node: torch.fx.Node) -> bool:
+        # Parameters are supported if any of there users are supported
+        if is_param_node(self.ep, node):
+            return any(
+                self.is_node_supported(submodules, user) for user in node.users.keys()
+            )
         # TODO - other ops?
         if node.op != "call_function":
             return False