Arm backend: Support TFA-decomposable INT ops in no-quant mixed profile (#20018)

In the INT-only profile, some integer ops are expected to be decomposed
by the transform-for-annotation pipeline before partitioning, so they
are intentionally absent from TOSA_PRO_INT_SupportList.

The no-quant mixed INT+FP profile does not run that pipeline, so the
original ops can reach the partitioner and be rejected because they are
absent from TOSA_PRO_INT_SupportList. However, these ops can still be
supported by decomposition passes in the backend pipeline.

Add a mixed INT support list that extends TOSA_PRO_INT_SupportList with
backend-decomposable integer ops for mixed-profile partitioning. Include
slice_scatter in that extension and remove the corresponding VGF
no-quant xfails

Change-Id: I0ccc5484dc8c8311cefb069df9e2a4878bd98c9a


cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils
@Sebastian-Larsson @robell @rascani

Signed-off-by: Yufeng Shi <yufeng.shi@arm.com>

Author: YufengShi-dudu

backends/arm/operator_support/tosa_profile_supported_op_lists.py

@@ -131,6 +131,22 @@
 }
 
 
+# Extra integer ops for the mixed INT+FP support list. These ops can be
+# supported by passes in the backend pipeline, but are intentionally kept out
+# of TOSA_PRO_INT_SupportList because INT-only partitioning expects them to be
+# decomposed before partitioning. Extend this list if the same mixed-profile
+# support gap is observed for other backend-decomposable ops.
+TOSA_PRO_MIXED_DECOMPOSABLE_INT_SupportList: Final[Set] = {
+    exir_ops.edge.aten.slice_scatter.default,
+}
+
+
+# INT-side support list used when partitioning under the mixed INT+FP profile.
+TOSA_PRO_MIXED_INT_SupportList: Final[Set] = (
+    TOSA_PRO_INT_SupportList | TOSA_PRO_MIXED_DECOMPOSABLE_INT_SupportList
+)
+
+
 # FP profile: ops supported via native TOSA ops, decompositions/transformations, precompute, etc.
 TOSA_PRO_FP_SupportList: Final[Set] = {
     exir_ops.edge.aten.abs.default,
@@ -257,5 +273,6 @@
 
 __all__ = [
     "TOSA_PRO_INT_SupportList",
+    "TOSA_PRO_MIXED_INT_SupportList",
     "TOSA_PRO_FP_SupportList",
 ]

backends/arm/operator_support/tosa_supported_operators.py

@@ -42,6 +42,7 @@
 from executorch.backends.arm.operator_support.tosa_profile_supported_op_lists import (
     TOSA_PRO_FP_SupportList,
     TOSA_PRO_INT_SupportList,
+    TOSA_PRO_MIXED_INT_SupportList,
 )
 from executorch.backends.arm.tosa.specification import (
     TosaSpecification,
@@ -453,7 +454,7 @@ def is_node_supported(
 
         # Select list based on whether the node is quantized.
         if is_quantized(node) or node.target in (*Q_OPS, *DQ_OPS):
-            support_list = TOSA_PRO_INT_SupportList
+            support_list = TOSA_PRO_MIXED_INT_SupportList
         else:
             support_list = TOSA_PRO_FP_SupportList
 

backends/arm/test/ops/test_slice_scatter.py

@@ -264,12 +264,6 @@ def test_slice_scatter_u85_INT_stepN(test_module: input_t):
 @common.parametrize(
     "test_module",
     test_data_int_step1 | test_data_int_stepN | test_data_fp_step1 | test_data_fp_stepN,
-    xfails={
-        "rank2_step1_int8": "MLETORCH-1823: Fix quantized-node detection",
-        "rank2_prefix_empty_int8": "MLETORCH-1823: Fix quantized-node detection",
-        "rank2_suffix_empty_end_none_int8": "MLETORCH-1823: Fix quantized-node detection",
-        "rank3_step2_int32": "MLETORCH-1823: Fix quantized-node detection",
-    },
 )
 def test_slice_scatter_vgf_no_quant(test_module: input_t):
     pipeline = VgfPipeline[input_t](