Update

[ghstack-poisoned]

Author: JCNTH

.github/workflows/add-unanswered-to-project.yml

@@ -27,11 +27,11 @@ jobs:
             // List of authors to exclude
             const excludedAuthors = new Set([
               "nil-is-all", "tanvirislam-meta", "cbilgin", "kimishpatel", "psiddh", "digantdesai", "SS-JIA", "ahmtox", "mcr229", 
-              "shoumikhin", "manuelcandales", "metascroy", "cccclai", "rohansjoshi", "kirklandsign", "abhinaykukkadapu", 
-              "JacobSzwejbka", "Conarnar", "rascani", "lucylq", "larryliu0820", "BujSet", "Gasoonjia", "Juntian777", "guangy10", 
+              "shoumikhin", "manuelcandales", "metascroy", "cccclai", "rohansjoshi", "kirklandsign", "abhinaykukkadapu", "JacobSzwejbka", 
+              "Conarnar", "rascani", "kiymetakdemir", "JCNTH", "lucylq", "larryliu0820", "BujSet", "Gasoonjia", "Juntian777", "guangy10", 
               "jackzhxng", "GregoryComer", "leafs1", "swolchok", "mergennachin", "tarun292", "byjlw", "jathu", "Jack-Khuu", "georgehong", 
-              "zhenyan-zhang-meta", "silverguo", "harishs88ss", "AlannaBurke", "dbort", "huydhn", "mcremon-meta", "trivedivivek", 
-              "angelayi", "helunwencser", "hsharma35", "zhxchen17", "iseeyuan", "svekars", "nathanaelsee", "dulinriley", 
+              "zhenyan-zhang-meta", "silverguo", "harishs88ss", "AlannaBurke", "Doggeral", "laithsakka", "Reubend", "dbort", "huydhn", "mcremon-meta", 
+              "trivedivivek", "angelayi", "helunwencser", "hsharma35", "zhxchen17", "iseeyuan", "svekars", "nathanaelsee", "dulinriley", 
               "jerryzh168", "cmodi-meta", "bigfootjon", "sxu", "ydwu4", "Riandy", "tugsbayasgalan", "bsoyluoglu", "yangw-dev", 
               "YIWENX14", "namanahuja", "yushangdi", "limintang", "pianpwk", "viveknayakatmeta", "andreanicastro", "JakeStevens",
               "gmagogsfm", "zonglinpeng", "eigen-k", "derekxu", "salilsdesai", "skrtskrtfb", "pssrawat", "r-barnes",
@@ -50,7 +50,7 @@ jobs:
               "mansnils", "martinlsm", "freddan80", "YufengShi-dudu", "tom-arm", "perheld", "Jerry-Ge", "gggekov", "fumchin", "wwwind", 
               "benkli01", "Tessil", "maddun01", "Michiel-Olieslagers", "armwaheed", "agrima1304", "emmakujala", "annietllnd", 
               "MatthiasHertel80", "AlexTawseArm", "jmahbs", "morgolock", "Christoffer-JL", "ArmRyan", "xingguo01", "tgonzalezorlandoarm", 
-              "chizkiyahu", "sarah-blades", "itsMarco-G", "usamahz", "Rob-Hughes-Arm", 
+              "chizkiyahu", "sarah-blades", "itsMarco-G", "usamahz", "Rob-Hughes-Arm", "swha815", "FabulousSuperDude",
               "haowhsu-quic", "shewu-quic", "winskuo-quic", "chunit-quic", "DannyYuyang-quic", "chuntl", "thchenqti", "jethroqti", 
               "chenweng-quic", "qti-horodnic", "qti-mmadhava", "quic-boyuc", "zhaoxul-qti", 
               "cymbalrush", "DenisVieriu97", "billmguo", 

backends/aoti/passes/TARGETS backends/aoti/passes/BUCKbackends/aoti/passes/TARGETS renamed to backends/aoti/passes/BUCK

@@ -1,11 +1,13 @@
+load("@fbcode_macros//build_defs:build_file_migration.bzl", "fbcode_target")
 load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
 
 oncall("executorch")
 
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "passes",
     srcs = [
-      "replace_view_copy_with_view.py",
+        "replace_view_copy_with_view.py",
     ],
     visibility = [
         "//executorch/...",

backends/arm/TARGETS backends/arm/BUCKbackends/arm/TARGETS renamed to backends/arm/BUCK

@@ -4,9 +4,11 @@
 # LICENSE file in the root directory of this source tree.
 
 # @noautodeps
+load("@fbcode_macros//build_defs:build_file_migration.bzl", "fbcode_target")
 load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
 
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "constants",
     srcs = [
         "constants.py",
@@ -15,7 +17,8 @@ runtime.python_library(
         "//executorch/exir/dialects:lib",
     ],
 )
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "ao_ext",
     srcs = glob([
         "ao_ext/*.py",
@@ -28,7 +31,8 @@ runtime.python_library(
     ],
 )
 
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "lib",
     srcs = [
         "__init__.py",
@@ -40,7 +44,8 @@ runtime.python_library(
         "//executorch/backends/arm/quantizer:lib",
     ],
 )
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "common",
     srcs = glob(["common/*.py"]),
     deps = [
@@ -50,7 +55,8 @@ runtime.python_library(
     ],
 )
 
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "arm_compile_spec",
     srcs = [
         "common/arm_compile_spec.py",
@@ -67,7 +73,8 @@ runtime.python_library(
         "//executorch/backends/arm/_passes:passes",
     ],
 )
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "ethosu",
     srcs = [
         "ethosu/__init__.py",
@@ -83,11 +90,13 @@ runtime.python_library(
     ],
 )
 
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "vgf",
     srcs = [
         "vgf/__init__.py",
         "vgf/_passes/__init__.py",
+        "vgf/_passes/insert_grid_sampler_grid_dequant_pass.py",
         "vgf/_passes/rewrite_grid_sampler_to_tosa_custom.py",
         "vgf/backend.py",
         "vgf/check_env.py",
@@ -119,7 +128,8 @@ runtime.python_library(
         "//executorch/exir:lib",
     ],
 )
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "public_api",
     srcs = ["__init__.py"],
     deps = [
@@ -129,7 +139,8 @@ runtime.python_library(
     ],
 )
 
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "process_node",
     srcs = ["process_node.py"],
     deps = [
@@ -141,7 +152,8 @@ runtime.python_library(
         "//executorch/exir:lib",
     ],
 )
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "arm_vela",
     srcs = [
         "arm_vela.py",
@@ -150,7 +162,8 @@ runtime.python_library(
         "fbsource//third-party/pypi/ethos-u-vela:ethos-u-vela",
     ],
 )
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "arm_model_evaluator",
     srcs = [
         "util/arm_model_evaluator.py",
@@ -159,7 +172,8 @@ runtime.python_library(
         "//caffe2:torch",
     ]
 )
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "_factory",
     srcs = [
         "util/_factory.py"

backends/arm/MODELS.md

@@ -5,6 +5,7 @@
 - Conformer
 - Deep AutoEncoder
 - Deit Tiny
+- DeepSeek-R1-Distill-Qwen-1.5B
 - DeepLab v3 (DL3)
 - DS CNN
 - Inception v3 (IC3)

backends/arm/_passes/TARGETS backends/arm/_passes/BUCKbackends/arm/_passes/TARGETS renamed to backends/arm/_passes/BUCK

@@ -1,6 +1,8 @@
+load("@fbcode_macros//build_defs:build_file_migration.bzl", "fbcode_target")
 load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
 
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "core",
     srcs = [
         "arm_pass.py",
@@ -17,7 +19,8 @@ runtime.python_library(
     ],
 )
 
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "arm_pass_manager_base",
     srcs = ["arm_pass_manager.py"],
     deps = [
@@ -32,7 +35,8 @@ runtime.python_library(
     ],
 )
 
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "arm_pass_manager_fb",
     srcs = [],
     deps = [
@@ -41,7 +45,8 @@ runtime.python_library(
     ],
 )
 
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "passes",
     srcs = glob(["*.py"], exclude = [
         "arm_pass.py",

backends/arm/_passes/arm_pass_manager.py

@@ -633,6 +633,9 @@ def _tosa_pipeline(
                 FuseViewCopyTransformPass(),
                 PropagateViewCopyPermuteDownPass(self.compile_spec, exported_program),
                 PropagateViewCopyPermuteUpPass(self.compile_spec, exported_program),
+                # Propagation can leave a binary op with mismatched operand ranks,
+                # which TOSA rejects; re-match ranks before lowering.
+                MatchArgRanksPass(exported_program),
                 RewriteHighRankSingletonPermutePass(),
                 DecomposePermuteForU55Pass(),
                 RewriteSlicePass(),

backends/arm/_passes/aten_to_tosa_data_layout.py

@@ -0,0 +1,98 @@
+# Copyright 2026 Arm Limited and/or its affiliates.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+from collections.abc import Sequence
+from typing import cast
+
+from executorch.backends.transforms.aten_to_dialect_pass import (
+    AtenToDialectPass,
+    DialectNodeSpec,
+)
+from executorch.exir.dialects._ops import ops as exir_ops
+from torch.fx import Node
+
+
+def _get_arg(node: Node, index: int, name: str, default=None):
+    if len(node.args) > index:
+        return node.args[index]
+    return node.kwargs.get(name, default)
+
+
+def _normalize_dim(dim: int, rank: int) -> int:
+    return (dim + rank) % rank
+
+
+def _input_rank(node: Node) -> int:
+    input_node = cast(Node, node.args[0])
+    return len(input_node.meta["val"].shape)
+
+
+def _rewrite_cat(node: Node, pass_: AtenToDialectPass) -> DialectNodeSpec:
+    tensors = cast(Sequence[Node], node.args[0])
+    dim = _get_arg(node, 1, "dim", 0)
+    first_tensor = tensors[0]
+    axis = _normalize_dim(cast(int, dim), len(first_tensor.meta["val"].shape))
+    return DialectNodeSpec(
+        exir_ops.backend.tosa.CONCAT.default,
+        (tensors,),
+        {"axis": axis},
+    )
+
+
+def _rewrite_view_copy(node: Node, pass_: AtenToDialectPass) -> DialectNodeSpec:
+    return DialectNodeSpec(
+        exir_ops.backend.tosa.RESHAPE.default,
+        node.args,
+        dict(node.kwargs),
+    )
+
+
+def _rewrite_repeat(node: Node, pass_: AtenToDialectPass) -> DialectNodeSpec:
+    return DialectNodeSpec(
+        exir_ops.backend.tosa.TILE.default,
+        node.args,
+        dict(node.kwargs),
+    )
+
+
+def _rewrite_permute_copy(node: Node, pass_: AtenToDialectPass) -> DialectNodeSpec:
+    permutation = list(cast(Sequence[int], _get_arg(node, 1, "dims")))
+    rank = _input_rank(node)
+    permutation = [_normalize_dim(dim, rank) for dim in permutation]
+    return DialectNodeSpec(
+        exir_ops.backend.tosa.TRANSPOSE.default,
+        (node.args[0], permutation),
+        {},
+    )
+
+
+def _rewrite_flip(node: Node, pass_: AtenToDialectPass) -> DialectNodeSpec | None:
+    dims = list(cast(Sequence[int], _get_arg(node, 1, "dims")))
+    if len(dims) != 1:
+        return None
+
+    return DialectNodeSpec(
+        exir_ops.backend.tosa.REVERSE.default,
+        (node.args[0],),
+        {"axis": _normalize_dim(dims[0], _input_rank(node))},
+    )
+
+
+def rewrite_data_layout_operator(
+    node: Node, pass_: AtenToDialectPass
+) -> DialectNodeSpec | None:
+    match node.target:
+        case exir_ops.edge.aten.cat.default:
+            return _rewrite_cat(node, pass_)
+        case exir_ops.edge.aten.view_copy.default:
+            return _rewrite_view_copy(node, pass_)
+        case exir_ops.edge.aten.repeat.default:
+            return _rewrite_repeat(node, pass_)
+        case exir_ops.edge.aten.permute_copy.default:
+            return _rewrite_permute_copy(node, pass_)
+        case exir_ops.edge.aten.flip.default:
+            return _rewrite_flip(node, pass_)
+        case _:
+            return None

backends/arm/_passes/exir_to_tosa_pass.py

@@ -9,6 +9,9 @@
 from executorch.backends.arm._passes.aten_to_tosa_activation_functions import (
     get_activation_replacement,
 )
+from executorch.backends.arm._passes.aten_to_tosa_data_layout import (
+    rewrite_data_layout_operator,
+)
 from executorch.backends.arm._passes.aten_to_tosa_tensor_operators import (
     rewrite_argmax,
     rewrite_binary_operator,
@@ -118,3 +121,16 @@ def _get_activation_replacement(
     node: Node, pass_: AtenToDialectPass
 ) -> DialectNodeSpec | None:
     return get_activation_replacement(node, pass_)
+
+
+@register_dialect_substitutions(
+    exir_ops.edge.aten.cat.default,
+    exir_ops.edge.aten.flip.default,
+    exir_ops.edge.aten.permute_copy.default,
+    exir_ops.edge.aten.repeat.default,
+    exir_ops.edge.aten.view_copy.default,
+)
+def _get_data_layout_replacement(
+    node: Node, pass_: AtenToDialectPass
+) -> DialectNodeSpec | None:
+    return rewrite_data_layout_operator(node, pass_)

backends/arm/_passes/insert_data_layout_casts_pass.py

@@ -35,7 +35,6 @@ class InsertDataLayoutCastsPass(ArmOpTargetedPass):
 
     _concat_ops = {
         exir_ops.edge.aten.cat.default,
-        exir_ops.edge.aten.concatenate.default,
         exir_ops.backend.tosa.CONCAT.default,
     }
     _single_input_ops = {

backends/arm/debug/TARGETS backends/arm/debug/BUCKbackends/arm/debug/TARGETS renamed to backends/arm/debug/BUCK

@@ -1,7 +1,9 @@
 # @noautodeps
+load("@fbcode_macros//build_defs:build_file_migration.bzl", "fbcode_target")
 load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
 
-runtime.python_library(
+fbcode_target(
+    _kind = runtime.python_library,
     name = "schema",
     srcs = [
         "__init__.py",