Skip to content

Commit fd318cc

Browse files
authored
Refactor source_transformation to a seperate target
Differential Revision: D69942050 Pull Request resolved: #8602
1 parent 254eeca commit fd318cc

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

examples/models/llama/TARGETS

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -95,11 +95,8 @@ runtime.command_alias(
9595
)
9696

9797
runtime.python_library(
98-
name = "export_library",
98+
name = "source_transformation",
9999
srcs = [
100-
"export_llama.py",
101-
"export_llama_lib.py",
102-
"model.py",
103100
"source_transformation/apply_spin_quant_r1_r2.py",
104101
"source_transformation/attention.py",
105102
"source_transformation/lora.py",
@@ -114,6 +111,15 @@ runtime.python_library(
114111
"source_transformation/vulkan_rope.py",
115112
"source_transformation/attention_sink.py",
116113
],
114+
)
115+
116+
runtime.python_library(
117+
name = "export_library",
118+
srcs = [
119+
"export_llama.py",
120+
"export_llama_lib.py",
121+
"model.py",
122+
],
117123
_is_external_target = True,
118124
base_module = "executorch.examples.models.llama",
119125
visibility = [
@@ -123,6 +129,7 @@ runtime.python_library(
123129
"@EXECUTORCH_CLIENTS",
124130
],
125131
deps = [
132+
":source_transformation",
126133
"//ai_codesign/gen_ai/fast_hadamard_transform:fast_hadamard_transform",
127134
"//caffe2:torch",
128135
"//executorch/backends/vulkan/_passes:vulkan_passes",

0 commit comments

Comments
 (0)