load("@fbcode_macros//build_defs:python_library.bzl", "python_library")

oncall("executorch")

python_library(
    name = "xnnpack_passes",
    srcs = [
        "__init__.py",
        "channels_last_tagged_reshape_pass.py",
        "conv1d_unsqueeze_pass.py",
        "convert_to_linear.py",
        "convert_to_sdpa.py",
        "convert_to_upsample_bilinear2d.py",
        "fuse_activation_pass.py",
        "fuse_batch_norm_with_conv.py",
        "prelu_reshape_pass.py",
        "remove_getitem_op.py",
        "tag_implicit_q_dq_pass.py",
        "xnnpack_pass.py",
    ],
    deps = [
        "//caffe2:torch",
        "//executorch/backends/transforms:addmm_mm_to_linear",
        "//executorch/backends/transforms:lib",
        "//executorch/backends/xnnpack/partition:configs",
        "//executorch/backends/xnnpack/partition:partitioner_graphs",
        "//executorch/backends/xnnpack/serialization:xnnpack_schema",
        "//executorch/backends/xnnpack/utils:xnnpack_utils",
        "//executorch/exir:lib",
        "//executorch/exir:pass_base",
        "//executorch/exir/dialects:lib",
        "//executorch/exir/passes:const_prop_pass",
        "//executorch/exir/passes:memory_format_ops_pass",
        "//executorch/exir/program:program",
    ],
)