Skip to content

Commit c07521a

Browse files
authored
Targets for Qualcomm llm eval
Differential Revision: D82655293 Pull Request resolved: #14381
1 parent d40ce3f commit c07521a

File tree

1 file changed

+12
-0
lines changed
  • examples/qualcomm/oss_scripts/llama

1 file changed

+12
-0
lines changed

examples/qualcomm/oss_scripts/llama/TARGETS

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,16 @@ runtime.python_library(
2626
],
2727
)
2828

29+
runtime.python_library(
30+
name = "masking_utils",
31+
srcs = [
32+
"masking_utils.py",
33+
],
34+
deps = [
35+
"//caffe2:torch",
36+
],
37+
)
38+
2939
runtime.python_library(
3040
name = "decoder_constants",
3141
srcs = [
@@ -39,6 +49,7 @@ runtime.python_library(
3949
deps = [
4050
":decoder_constants",
4151
":decoder_utils",
52+
":masking_utils",
4253
"//executorch/examples/models/llama:source_transformation",
4354
"//caffe2:torch",
4455
"//executorch/backends/qualcomm/partition:partition",
@@ -90,6 +101,7 @@ python_binary(
90101
"//executorch/examples/qualcomm/oss_scripts/llama:range_setting_pt2e",
91102
"fbsource//third-party/pypi/lm-eval:lm-eval",
92103
],
104+
keep_gpu_sections = True,
93105
)
94106

95107
runtime.command_alias(

0 commit comments

Comments
 (0)