fix mismatch sub dtype (#13447)

Summary: Pull Request resolved: #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,