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 1515
1616import torchgen
1717from executorch .codegen .api .custom_ops import ComputeNativeFunctionStub
18+ from executorch .codegen .gen import gen_headers
1819from executorch .codegen .model import ETKernelIndex
19- from torchgen .gen_executorch import gen_headers
2020from torchgen .model import Location , NativeFunction
2121from torchgen .selective_build .selector import SelectiveBuilder
2222from torchgen .utils import FileManager
Original file line number Diff line number Diff line change 1111import unittest
1212
1313import 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 (
1815 ComputeCodegenUnboxedKernels ,
1916 gen_functions_declarations ,
2017 parse_yaml_files ,
2118 translate_native_yaml ,
2219)
20+
21+ from executorch .codegen .model import ETKernelIndex , ETKernelKey
22+ from torchgen .gen import LineLoader
2323from torchgen .model import (
2424 BackendIndex ,
2525 BackendMetadata ,
Original file line number Diff line number Diff line change @@ -81,18 +81,6 @@ runtime.python_binary(
8181 _is_external_target = True,
8282)
8383
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-
9684runtime.filegroup(
9785 name = "aten_src_path",
9886 srcs = [
Original file line number Diff line number Diff line change @@ -91,8 +91,9 @@ function(generate_bindings_for_kernels)
9191 OUTPUT_STRIP_TRAILING_WHITESPACE
9292 )
9393 file (GLOB_RECURSE _torchgen_srcs "${torchgen-out}/*.py" )
94+ # Not using module executorch.codegen.gen because it's not installed yet.
9495 set (_gen_command
95- "${PYTHON_EXECUTABLE} " -m torchgen.gen_executorch
96+ "${PYTHON_EXECUTABLE} " ${EXECUTORCH_ROOT} /codegen/gen.py
9697 --source -path =${EXECUTORCH_ROOT} /codegen --install -dir=${_out_dir}
9798 --tags-path =${torchgen-out}/packaged/ATen/native/tags.yaml
9899 --aten-yaml-path =${torchgen-out}/packaged/ATen/native/native_functions.yaml
You can’t perform that action at this time.
0 commit comments