File tree Expand file tree Collapse file tree 4 files changed +7
-18
lines changed Expand file tree Collapse file tree 4 files changed +7
-18
lines changed Original file line number Diff line number Diff line change 15
15
16
16
import torchgen
17
17
from executorch .codegen .api .custom_ops import ComputeNativeFunctionStub
18
+ from executorch .codegen .gen import gen_headers
18
19
from executorch .codegen .model import ETKernelIndex
19
- from torchgen .gen_executorch import gen_headers
20
20
from torchgen .model import Location , NativeFunction
21
21
from torchgen .selective_build .selector import SelectiveBuilder
22
22
from torchgen .utils import FileManager
Original file line number Diff line number Diff line change 11
11
import unittest
12
12
13
13
import yaml
14
-
15
- from executorch .codegen .model import ETKernelIndex , ETKernelKey
16
- from torchgen .gen import LineLoader
17
- from torchgen .gen_executorch import (
14
+ from executorch .codegen .gen import (
18
15
ComputeCodegenUnboxedKernels ,
19
16
gen_functions_declarations ,
20
17
parse_yaml_files ,
21
18
translate_native_yaml ,
22
19
)
20
+
21
+ from executorch .codegen .model import ETKernelIndex , ETKernelKey
22
+ from torchgen .gen import LineLoader
23
23
from torchgen .model import (
24
24
BackendIndex ,
25
25
BackendMetadata ,
Original file line number Diff line number Diff line change @@ -81,18 +81,6 @@ runtime.python_binary(
81
81
_is_external_target = True,
82
82
)
83
83
84
- runtime.python_binary(
85
- name = "gen_executorch",
86
- main_module = "torchgen.gen_executorch",
87
- visibility = [
88
- "PUBLIC",
89
- ],
90
- deps = [
91
- ":torchgen",
92
- ],
93
- _is_external_target = True,
94
- )
95
-
96
84
runtime.filegroup(
97
85
name = "aten_src_path",
98
86
srcs = [
Original file line number Diff line number Diff line change @@ -91,8 +91,9 @@ function(generate_bindings_for_kernels)
91
91
OUTPUT_STRIP_TRAILING_WHITESPACE
92
92
)
93
93
file (GLOB_RECURSE _torchgen_srcs "${torchgen-out}/*.py" )
94
+ # Not using module executorch.codegen.gen because it's not installed yet.
94
95
set (_gen_command
95
- "${PYTHON_EXECUTABLE} " -m torchgen.gen_executorch
96
+ "${PYTHON_EXECUTABLE} " ${EXECUTORCH_ROOT} /codegen/gen.py
96
97
--source-path=${EXECUTORCH_ROOT}/codegen --install-dir=${_out_dir}
97
98
--tags-path=${torchgen-out}/packaged/ATen/native/tags.yaml
98
99
--aten-yaml-path=${torchgen-out}/packaged/ATen/native/native_functions.yaml
You can’t perform that action at this time.
0 commit comments