diff --git a/backends/vulkan/targets.bzl b/backends/vulkan/targets.bzl index b9b96abdec4..775341d420d 100644 --- a/backends/vulkan/targets.bzl +++ b/backends/vulkan/targets.bzl @@ -387,6 +387,8 @@ def define_common_targets(is_fbcode = False): "//executorch/backends/transforms:view_copy_to_squeeze_unsqueeze", "//executorch/backends/vulkan/_passes:vulkan_passes", "//executorch/backends/vulkan/serialization:lib", + "//executorch/backends/transforms:remove_getitem_op", + "//executorch/backends/xnnpack/_passes:xnnpack_passes", "//executorch/exir/backend:backend_details", ], ) diff --git a/backends/vulkan/vulkan_preprocess.py b/backends/vulkan/vulkan_preprocess.py index 1816d9b12de..5db5d7a4ff4 100644 --- a/backends/vulkan/vulkan_preprocess.py +++ b/backends/vulkan/vulkan_preprocess.py @@ -13,9 +13,6 @@ import executorch.backends.vulkan.utils as utils from executorch.backends.transforms.addmm_mm_to_linear import AddmmToLinearTransform -from executorch.backends.transforms.fuse_batch_norm_with_conv import ( - FuseBatchNormWithConvPass, -) from executorch.backends.transforms.fuse_conv_with_clamp import FuseClampPass from executorch.backends.transforms.fuse_view_copy import FuseViewCopyTransform from executorch.backends.transforms.view_copy_to_squeeze_unsqueeze import ( @@ -40,6 +37,7 @@ from executorch.backends.vulkan.serialization.vulkan_graph_serialize import ( serialize_vulkan_graph, ) +from executorch.backends.xnnpack._passes import FuseBatchNormPass from executorch.exir.backend.backend_details import ( BackendDetails, @@ -162,7 +160,7 @@ def preprocess( # noqa: C901 SqueezeUnsqueezeInputs(), FuseViewCopyTransform(), ViewCopyToSqueezeUnsqueezePass(), - FuseBatchNormWithConvPass(program), + FuseBatchNormPass(program), FuseClampPass(), ], )