1
- load ("@fbsource//xplat/executorch/build:runtime_wrapper.bzl" , "get_oss_build_kwargs" , "runtime" )
1
+ load ("@fbsource//xplat/executorch/build:runtime_wrapper.bzl" , "get_aten_mode_options" , " get_oss_build_kwargs" , "runtime" )
2
2
load ("@fbsource//xplat/executorch/codegen:codegen.bzl" , "executorch_generated_lib" )
3
3
4
4
def define_common_targets ():
@@ -10,23 +10,27 @@ def define_common_targets():
10
10
11
11
# Wraps a commandline executable that can be linked against any desired
12
12
# kernel or backend implementations. Contains a main() function.
13
- runtime .cxx_library (
14
- name = "executor_runner_lib" ,
15
- srcs = ["executor_runner.cpp" ],
16
- deps = [
17
- "//executorch/runtime/executor:program" ,
18
- "//executorch/extension/data_loader:file_data_loader" ,
19
- "//executorch/extension/evalue_util:print_evalue" ,
20
- "//executorch/extension/runner_util:inputs" ,
21
- ],
22
- external_deps = [
23
- "gflags" ,
24
- ],
25
- define_static_target = True ,
26
- visibility = [
27
- "//executorch/examples/..." ,
28
- ],
29
- )
13
+ for aten_mode in get_aten_mode_options ():
14
+ aten_suffix = "_aten" if aten_mode else ""
15
+
16
+ runtime .cxx_library (
17
+ name = "executor_runner_lib" ,
18
+ srcs = ["executor_runner.cpp" ],
19
+ deps = [
20
+ "//executorch/runtime/executor:program" ,
21
+ "//executorch/devtools/etdump:etdump_flatcc" + aten_suffix ,
22
+ "//executorch/extension/data_loader:file_data_loader" ,
23
+ "//executorch/extension/evalue_util:print_evalue" ,
24
+ "//executorch/extension/runner_util:inputs" ,
25
+ ],
26
+ external_deps = [
27
+ "gflags" ,
28
+ ],
29
+ define_static_target = True ,
30
+ visibility = [
31
+ "//executorch/examples/..." ,
32
+ ],
33
+ )
30
34
31
35
runtime .cxx_library (
32
36
name = "executor_runner_lib_with_threadpool" ,
0 commit comments