Skip to content

Commit 53b99f3

Browse files
larryliu0820facebook-github-bot
authored andcommitted
Refactor llama_runner to use Module (#1488)
Summary: Pull Request resolved: #1488 As titled, address comments on D52208223 Reviewed By: shoumikhin Differential Revision: D52423347 fbshipit-source-id: 85f14cbaa8d13971838dd3058d2282a33fb373ee
1 parent ff28629 commit 53b99f3

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

extension/runner/module/targets.bzl

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,3 +24,21 @@ def define_common_targets():
2424
"//executorch/extension/runner:runner",
2525
],
2626
)
27+
28+
runtime.cxx_library(
29+
name = "module_aten",
30+
srcs = [
31+
"module.cpp",
32+
],
33+
exported_headers = [
34+
"module.h",
35+
],
36+
visibility = [
37+
"@EXECUTORCH_CLIENTS",
38+
],
39+
exported_deps = [
40+
"//executorch/extension/memory_allocator:malloc_memory_allocator",
41+
"//executorch/extension/data_loader:mmap_data_loader",
42+
"//executorch/extension/runner:runner_aten",
43+
],
44+
)

0 commit comments

Comments
 (0)