executorch/devtools/inspector/TARGETS at main · apullin/executorch · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
load("@fbcode_macros//build_defs:python_binary.bzl", "python_binary")
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")

oncall("executorch")

runtime.python_library(
    name = "inspector",
    srcs = [
        "_inspector.py",
    ],
    deps = [
        "fbsource//third-party/pypi/ipython:ipython",
        "fbsource//third-party/pypi/numpy:numpy",
        "fbsource//third-party/pypi/pandas:pandas",
        "fbsource//third-party/pypi/tabulate:tabulate",
        ":inspector_utils",
        "//executorch/devtools/debug_format:et_schema",
        "//executorch/devtools/etdump:schema_flatcc",
        "//executorch/devtools/etrecord:etrecord",
        "//executorch/exir:lib",
        "//executorch/devtools/inspector:intermediate_output_capturer",
        "//executorch/devtools/inspector/numerical_comparator:lib",
    ],
)

python_binary(
    name = "inspector_cli",
    main_function = ".inspector_cli.main",
    main_src = "inspector_cli.py",
    deps = [
        "//executorch/devtools:lib",
        "//executorch/devtools/inspector:lib",
    ],
)

runtime.python_library(
    name = "inspector_utils",
    srcs = [
        "_inspector_utils.py",
    ],
    deps = [
        "fbsource//third-party/pypi/matplotlib:matplotlib",
        "fbsource//third-party/pypi/numpy:numpy",
        "//caffe2:torch",
        "//executorch/devtools/debug_format:base_schema",
        "//executorch/devtools/debug_format:et_schema",
        "//executorch/devtools/etdump:schema_flatcc",
        "//executorch/devtools/etdump:serialize",
        "//executorch/devtools/etrecord:etrecord",
    ],
)

runtime.python_library(
    name = "intermediate_output_capturer",
    srcs = [
        "_intermediate_output_capturer.py",
    ],
    deps = [
        "//executorch/devtools/inspector:inspector_utils",
    ],
)

runtime.python_library(
    name = "lib",
    srcs = ["__init__.py"],
    deps = [
        ":inspector",
        ":inspector_utils",
    ],
)