Stop SymInt from coming in (#1778)

digantdesai · facebook-github-bot · commit aff6f681540a · 2024-01-31T14:40:44.000-08:00
Summary: Pull Request resolved: #1778 Drop an op with a non-tensor arg. This can be improved upon later. Reviewed By: mcr229 Differential Revision: D53248930 fbshipit-source-id: 004e6dd420a9d6c9753aca96c380033e650f3a54
diff --git a/backends/xnnpack/partition/xnnpack_partitioner.py b/backends/xnnpack/partition/xnnpack_partitioner.py
@@ -125,6 +125,11 @@ def check_node_has_valid_dtype(self, node):
             if arg_val is None or isinstance(arg_val, tuple):
                 continue
 
+            # Being conservative for now, UX >> Perf
+            # TODO: We need a pass to scrub these out.
+            if not isinstance(arg_val, torch.Tensor):
+                return False
+
             if arg_val.dtype not in valid_dtypes:
                 return False
 
@@ -135,6 +140,9 @@ def check_node_has_valid_dtype(self, node):
                 node_val = (node_val,)
 
             for val in node_val:
+                if not isinstance(val, torch.Tensor):
+                    return False
+
                 if val.dtype not in valid_dtypes:
                     return False
 
