fix mismatch sub dtype (pytorch#13447)

Summary: Pull Request resolved: pytorch#13447

Reviewed By: SS-JIA, digantdesai

Differential Revision: D80312352

Author: cccclai

exir/passes/remove_mixed_type_operators.py

@@ -23,6 +23,7 @@ def call_operator(self, op, args, kwargs, meta: NodeMetadata):  # noqa: C901
         promotion_type_allow_list = {
             torch.ops.aten.add.Tensor: ELEMENTWISE_TYPE_PROMOTION_KIND.DEFAULT,
             torch.ops.aten.mul.Tensor: ELEMENTWISE_TYPE_PROMOTION_KIND.DEFAULT,
+            torch.ops.aten.sub.Tensor: ELEMENTWISE_TYPE_PROMOTION_KIND.DEFAULT,
             # The correct promotion for div depends on the mode! If there is no mode,
             # it's INT_TO_FLOAT, otherwise it's default.
             torch.ops.aten.div.Tensor: ELEMENTWISE_TYPE_PROMOTION_KIND.INT_TO_FLOAT,

exir/tests/test_passes.py

@@ -159,6 +159,7 @@ def forward(self, x: torch.Tensor, y: torch.Tensor) -> torch.Tensor:
             return Module
 
         Add = make_module(lambda x, y: (x + y) + x)
+        Sub = make_module(lambda x, y: (x - y) - x)
         Mult = make_module(lambda x, y: x * y)
         Minimum = make_module(torch.minimum)
         DivWithoutMode = make_module(torch.div)
@@ -177,6 +178,12 @@ def forward(self, x: torch.Tensor, y: torch.Tensor) -> torch.Tensor:
                 2,
                 ELEMENTWISE_TYPE_PROMOTION_KIND.DEFAULT,
             ),
+            (
+                Sub,
+                exir_ops.edge.aten.sub.Tensor,
+                2,
+                ELEMENTWISE_TYPE_PROMOTION_KIND.DEFAULT,
+            ),
             (
                 Mult,
                 exir_ops.edge.aten.mul.Tensor,
@@ -998,9 +1005,9 @@ def test_debug_handle_generator_pass_generate_same_debug_handle_on_ops_sharing_s
                     for node_name_with_same_debug_handle in same_source_nodes[
                         node.name
                     ]:
-                        node_name_to_debug_handle[node_name_with_same_debug_handle] = (
-                            node.meta["debug_handle"]
-                        )
+                        node_name_to_debug_handle[
+                            node_name_with_same_debug_handle
+                        ] = node.meta["debug_handle"]
 
     def test_generate_missing_debug_handles(self) -> None:
         eager_model = MLP(2, output_size=4)