Skip to content

Commit ca32105

Browse files
authored
[executorch][flat_tensor] Generate flatc files (#7521)
Pull Request resolved: #7255 1. Move schema files under flat_tensor/serialize 2. Add targets to generate schema files ghstack-source-id: 260014922 @exported-using-ghexport Differential Revision: [D66903492](https://our.internmc.facebook.com/intern/diff/D66903492/)
1 parent 9fca4d8 commit ca32105

File tree

5 files changed

+54
-2
lines changed

5 files changed

+54
-2
lines changed
+16
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
2+
load(":targets.bzl", "define_common_targets")
3+
4+
oncall("executorch")
5+
6+
define_common_targets()
7+
8+
runtime.python_library(
9+
name = "schema",
10+
srcs = [
11+
"flat_tensor_schema.py",
12+
],
13+
visibility = [
14+
"//executorch/...",
15+
],
16+
)

extension/flat_tensor/flat_tensor.fbs renamed to extension/flat_tensor/serialize/flat_tensor.fbs

+1-1
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ table TensorMetadata {
1313
scalar_type: executorch_flatbuffer.ScalarType;
1414

1515
// Size of each dimension.
16-
dim_sizes: [int32];
16+
sizes: [int32];
1717

1818
// Specifies in what order the dimensions are laid out in memory (from outer
1919
// to inner).

extension/flat_tensor/flat_tensor_schema.py renamed to extension/flat_tensor/serialize/flat_tensor_schema.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
class TensorMetadata:
1818
fully_qualified_name: str
1919
scalar_type: ScalarType
20-
dim_sizes: List[int]
20+
sizes: List[int]
2121
dim_order: List[bytes]
2222

2323
segment_index: int
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime")
2+
3+
def define_common_targets():
4+
runtime.genrule(
5+
name = "gen_schema",
6+
srcs = [
7+
"flat_tensor.fbs",
8+
"scalar_type.fbs",
9+
],
10+
outs = {
11+
"schema_generated.h": ["flat_tensor_generated.h"],
12+
"scalar_type_generated.h": ["scalar_type_generated.h"]
13+
},
14+
cmd = " ".join([
15+
"$(exe {})".format(runtime.external_dep_location("flatc")),
16+
"--cpp",
17+
"--cpp-std c++11",
18+
"--scoped-enums",
19+
"-o ${OUT}",
20+
"${SRCS}",
21+
]),
22+
default_outs = ["."],
23+
)
24+
25+
runtime.cxx_library(
26+
name = "generated_headers",
27+
srcs = [],
28+
visibility = [
29+
"//executorch/...",
30+
],
31+
exported_headers = {
32+
"schema_generated.h": ":gen_schema[schema_generated.h]",
33+
"scalar_type_generated.h": ":gen_schema[scalar_type_generated.h]",
34+
},
35+
exported_external_deps = ["flatbuffers-api"],
36+
)

0 commit comments

Comments
 (0)