Skip to content

Commit b5ea516

Browse files
Gasoonjiafacebook-github-bot
authored andcommitted
move all runtime apis under bundled_program/runtime.cpp/.h (#1050)
Summary: Pull Request resolved: #1050 Pull Request resolved: #434 This stack focuses on moving all bundled program codes under executorch/sdk/bundled_program. Target and discussions can be found in D49605947. This diff moves all runtime apis under bundled_program/runtime.cpp and bundled_program/runtime.h, and update dependencies. Reviewed By: tarun292 Differential Revision: D50422561 fbshipit-source-id: 2355af4ea6425f59f68d3f5947173b50f1eda7ac
1 parent 9fe1b3a commit b5ea516

File tree

12 files changed

+59
-30
lines changed

12 files changed

+59
-30
lines changed

examples/apple/mps/executor_runner/mps_executor_runner.mm

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
#include <executorch/runtime/platform/profiler.h>
2828
#include <executorch/runtime/platform/runtime.h>
2929
#include <executorch/util/util.h>
30-
#include <executorch/util/bundled_program_verification.h>
30+
#include <executorch/sdk/bundled_program/bundled_program.h>
3131
#include <executorch/extension/data_loader/buffer_data_loader.h>
3232
#include <executorch/runtime/core/result.h>
3333
#include <executorch/runtime/platform/runtime.h>

examples/apple/mps/executor_runner/targets.bzl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def define_common_targets():
2020
"//executorch/extension/data_loader:file_data_loader",
2121
"//executorch/extension/data_loader:buffer_data_loader",
2222
"//executorch/util:util",
23-
"//executorch/util:bundled_program_verification",
23+
"//executorch/sdk/bundled_program:runtime",
2424
"//executorch/util:util",
2525
],
2626
define_static_target = True,

examples/sdk/sdk_example_runner/sdk_example_runner.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,8 @@
2727
#include <executorch/runtime/executor/program.h>
2828
#include <executorch/runtime/platform/log.h>
2929
#include <executorch/runtime/platform/runtime.h>
30+
#include <executorch/sdk/bundled_program/bundled_program.h>
3031
#include <executorch/sdk/etdump/etdump_flatcc.h>
31-
#include <executorch/util/bundled_program_verification.h>
3232
#include <executorch/util/util.h>
3333

3434
static uint8_t method_allocator_pool[4 * 1024U * 1024U]; // 4MB

examples/sdk/sdk_example_runner/targets.bzl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@ def define_common_targets():
2020
"//executorch/extension/data_loader:file_data_loader",
2121
"//executorch/extension/data_loader:buffer_data_loader",
2222
"//executorch/util:util",
23-
"//executorch/util:bundled_program_verification",
2423
"//executorch/sdk/etdump:etdump_flatcc",
24+
"//executorch/sdk/bundled_program:runtime",
2525
],
2626
external_deps = [
2727
"gflags",

extension/pybindings/pybindings.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
#include <executorch/runtime/platform/profiler.h>
2727
#include <executorch/runtime/platform/runtime.h>
2828
#include <executorch/schema/bundled_program_schema_generated.h>
29-
#include <executorch/util/bundled_program_verification.h>
29+
#include <executorch/sdk/bundled_program/bundled_program.h>
3030
#include <executorch/util/read_file.h>
3131

3232
#include <ATen/Functions.h>

schema/targets.bzl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,7 @@ def define_common_targets():
9595
# Lock this down as tightly as possible to ensure that flatbuffers
9696
# are an implementation detail. Ideally this list would only include
9797
# //executorch/runtime/executor/...
98+
"//executorch/bundled_program/runtime/...",
9899
"//executorch/codegen/tools/...",
99100
"//executorch/runtime/executor/...",
100101
],

sdk/bundled_program/TARGETS

Lines changed: 22 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,19 @@
1-
load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
1+
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
2+
load(":targets.bzl", "define_common_targets")
23

34
oncall("executorch")
45

5-
python_library(
6+
define_common_targets()
7+
8+
runtime.python_library(
69
name = "core",
710
srcs = [
811
"core.py",
912
],
13+
visibility = [
14+
"//executorch/...",
15+
"@EXECUTORCH_CLIENTS",
16+
],
1017
deps = [
1118
":config",
1219
":schema",
@@ -18,30 +25,40 @@ python_library(
1825
],
1926
)
2027

21-
python_library(
28+
runtime.python_library(
2229
name = "config",
2330
srcs = [
2431
"config.py",
2532
],
33+
visibility = [
34+
"//executorch/...",
35+
"@EXECUTORCH_CLIENTS",
36+
],
2637
deps = [
2738
"fbsource//third-party/pypi/typing-extensions:typing-extensions",
2839
"//caffe2:torch",
2940
],
3041
)
3142

32-
python_library(
43+
runtime.python_library(
3344
name = "schema",
3445
srcs = [
3546
"schema.py",
3647
],
48+
visibility = [
49+
"//executorch/sdk/...",
50+
],
3751
deps = [
3852
"//executorch/exir:scalar_type",
3953
],
4054
)
4155

42-
python_library(
56+
runtime.python_library(
4357
name = "version",
4458
srcs = [
4559
"version.py",
4660
],
61+
visibility = [
62+
"//executorch/sdk/...",
63+
],
4764
)

util/bundled_program_verification.cpp renamed to sdk/bundled_program/bundled_program.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
* LICENSE file in the root directory of this source tree.
77
*/
88

9-
#include <executorch/util/bundled_program_verification.h>
9+
#include <executorch/sdk/bundled_program/bundled_program.h>
1010

1111
#include <cmath>
1212
#include <cstddef>

sdk/bundled_program/targets.bzl

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
2+
3+
def define_common_targets():
4+
"""Defines targets that should be shared between fbcode and xplat.
5+
6+
The directory containing this targets.bzl file should also contain both
7+
TARGETS and BUCK files that call this function.
8+
"""
9+
10+
for aten_mode in (True, False):
11+
aten_suffix = ("_aten" if aten_mode else "")
12+
runtime.cxx_library(
13+
name = "runtime" + aten_suffix,
14+
srcs = ["bundled_program.cpp"],
15+
exported_headers = ["bundled_program.h"],
16+
visibility = [
17+
"//executorch/...",
18+
"@EXECUTORCH_CLIENTS",
19+
],
20+
deps = [
21+
"//executorch/runtime/core/exec_aten/util:dim_order_util" + aten_suffix,
22+
"//executorch/schema:bundled_program_schema",
23+
],
24+
exported_deps = [
25+
"//executorch/runtime/core:memory_allocator",
26+
"//executorch/runtime/executor:program" + aten_suffix,
27+
],
28+
)

shim/xplat/executorch/extension/pybindings/pybindings.bzl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ PORTABLE_MODULE_DEPS = [
1414
"//executorch/runtime/executor:program",
1515
"//executorch/schema:bundled_program_schema",
1616
"//executorch/extension/aten_util:aten_bridge",
17-
"//executorch/util:bundled_program_verification",
17+
"//executorch/sdk/bundled_program:runtime",
1818
"//executorch/extension/data_loader:buffer_data_loader",
1919
"//executorch/extension/data_loader:mmap_data_loader",
2020
"//executorch/extension/memory_allocator:malloc_memory_allocator",
@@ -31,7 +31,7 @@ ATEN_MODULE_DEPS = [
3131
"//executorch/extension/data_loader:mmap_data_loader",
3232
"//executorch/extension/memory_allocator:malloc_memory_allocator",
3333
"//executorch/util:read_file",
34-
"//executorch/util:bundled_program_verification_aten",
34+
"//executorch/sdk/bundled_program:runtime_aten",
3535
"//caffe2:torch",
3636
"//caffe2:torch_extension",
3737
"//caffe2:ATen",

util/targets.bzl

Lines changed: 0 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -23,23 +23,6 @@ def define_common_targets():
2323

2424
for aten_mode in (True, False):
2525
aten_suffix = ("_aten" if aten_mode else "")
26-
runtime.cxx_library(
27-
name = "bundled_program_verification" + aten_suffix,
28-
srcs = ["bundled_program_verification.cpp"],
29-
exported_headers = ["bundled_program_verification.h"],
30-
visibility = [
31-
"//executorch/...",
32-
"@EXECUTORCH_CLIENTS",
33-
],
34-
deps = [
35-
"//executorch/runtime/core/exec_aten/util:dim_order_util" + aten_suffix,
36-
"//executorch/schema:bundled_program_schema",
37-
],
38-
exported_deps = [
39-
"//executorch/runtime/core:memory_allocator",
40-
"//executorch/runtime/executor:program" + aten_suffix,
41-
],
42-
)
4326

4427
runtime.cxx_library(
4528
name = "util" + aten_suffix,

0 commit comments

Comments
 (0)