diff --git a/backends/arm/_passes/TARGETS b/backends/arm/_passes/TARGETS index 7f61b6c2b4d..02d8549ac85 100644 --- a/backends/arm/_passes/TARGETS +++ b/backends/arm/_passes/TARGETS @@ -7,6 +7,7 @@ python_library( "//executorch/backends/arm:tosa_quant_utils", "//executorch/backends/arm:tosa_utils", "//executorch/backends/transforms:fuse_view_copy", + "//executorch/backends/transforms:remove_getitem_op", "//executorch/backends/transforms:replace_scalar_with_tensor", "//executorch/backends/xnnpack/_passes:xnnpack_passes", "//executorch/exir:lib", diff --git a/backends/arm/operator_support/TARGETS b/backends/arm/operator_support/TARGETS index c0c5af7487b..e14552fd016 100644 --- a/backends/arm/operator_support/TARGETS +++ b/backends/arm/operator_support/TARGETS @@ -6,6 +6,7 @@ python_library( deps = [ "//executorch/backends/arm/_passes:passes", "//executorch/backends/arm:tosa_specification", + "//executorch/backends/transforms:remove_getitem_op", "//executorch/backends/xnnpack/_passes:xnnpack_passes", "//executorch/exir:lib", ], diff --git a/backends/transforms/targets.bzl b/backends/transforms/targets.bzl index 71980195962..ad6d93420e3 100644 --- a/backends/transforms/targets.bzl +++ b/backends/transforms/targets.bzl @@ -117,6 +117,19 @@ def define_common_targets(): ], ) + runtime.python_library( + name = "remove_getitem_op", + srcs = ["remove_getitem_op.py"], + visibility = [ + "//executorch/backends/...", + ], + deps = [ + "//caffe2:torch", + "//executorch/exir:pass_base", + "//executorch/exir/dialects:lib", + ], + ) + runtime.python_library( name = "mean_to_sum_div", srcs = ["mean_to_sum_div.py"], diff --git a/backends/xnnpack/TARGETS b/backends/xnnpack/TARGETS index 4a3dfed7625..d5c6d6303d2 100644 --- a/backends/xnnpack/TARGETS +++ b/backends/xnnpack/TARGETS @@ -16,6 +16,7 @@ runtime.python_library( ], deps = [ "//executorch/backends/transforms:lib", + "//executorch/backends/transforms:remove_getitem_op", "//executorch/backends/xnnpack/_passes:xnnpack_passes", "//executorch/backends/xnnpack/operators:operators", "//executorch/backends/xnnpack/serialization:xnnpack_serializer", diff --git a/backends/xnnpack/test/TARGETS b/backends/xnnpack/test/TARGETS index 9b2ce0a4e82..bd3dddd0985 100644 --- a/backends/xnnpack/test/TARGETS +++ b/backends/xnnpack/test/TARGETS @@ -13,6 +13,7 @@ runtime.python_test( "test_xnnpack_utils_classes.py", ], deps = [ + "//executorch/backends/transforms:remove_getitem_op", "//executorch/backends/xnnpack/_passes:xnnpack_passes", "//executorch/backends/xnnpack/test/tester:tester", "//executorch/backends/xnnpack/utils:xnnpack_utils",