# @noautodeps

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

oncall("executorch")

python_library(
    name = "exported_module",
    srcs = [
        "exported_module.py",
    ],
    visibility = [
        "//executorch/...",
    ],
    deps = [
        "//caffe2:torch",
        "//executorch/exir:dynamic_shape",
        "//executorch/exir:lib",
        "//executorch/exir:pass_manager",
        "//executorch/exir/passes:lib",
    ],
)

python_library(
    name = "register_scratch_meta_fns",
    srcs = [
        "register_scratch_meta_fns.py",
    ],
    deps = [
        "//executorch/exir/operator:manip",
    ],
)

python_unittest(
    name = "end2end_aten",
    srcs = [
        "test_end2end.py",
    ],
    preload_deps = ["//executorch/kernels/portable:custom_ops_generated_lib"],
    deps = [
        ":exported_module",
        ":register_scratch_meta_fns",
        "//caffe2:torch",
        "//executorch/devtools:lib",
        "//executorch/devtools/bundled_program:config",
        "//executorch/devtools/bundled_program/serialize:lib",
        "//executorch/exir:dynamic_shape",
        "//executorch/exir:lib",
        "//executorch/exir:memory",
        "//executorch/exir:pass_manager",
        "//executorch/exir:print_program",
        "//executorch/exir:tensor",
        "//executorch/exir/_serialize:lib",
        "//executorch/exir/emit:lib",
        "//executorch/exir/passes:lib",
        "//executorch/exir/tests:control_flow_models",
        "//executorch/exir/tests:dynamic_shape_models",
        "//executorch/exir/tests:models",
        "//executorch/exir/tests:transformer",
        "//executorch/extension/pybindings:aten_lib",
        "//executorch/extension/pytree:pybindings",
    ],
)

python_unittest(
    name = "end2end",
    srcs = [
        "test_end2end.py",
    ],
    preload_deps = ["//executorch/kernels/portable:custom_ops_generated_lib"],
    deps = [
        ":exported_module",
        ":register_scratch_meta_fns",
        "//caffe2:torch",
        "//executorch/devtools:lib",
        "//executorch/devtools/bundled_program:config",
        "//executorch/devtools/bundled_program/serialize:lib",
        "//executorch/exir:dynamic_shape",
        "//executorch/exir:lib",
        "//executorch/exir:memory",
        "//executorch/exir:pass_manager",
        "//executorch/exir:print_program",
        "//executorch/exir:tensor",
        "//executorch/exir/_serialize:lib",
        "//executorch/exir/emit:lib",
        "//executorch/exir/passes:lib",
        "//executorch/exir/tests:control_flow_models",
        "//executorch/exir/tests:dynamic_shape_models",
        "//executorch/exir/tests:models",
        "//executorch/exir/tests:transformer",
        "//executorch/extension/pybindings:portable_lib",
        "//executorch/extension/pytree:pybindings",
    ],
)