File tree Expand file tree Collapse file tree 1 file changed +12
-0
lines changed
examples/qualcomm/oss_scripts/llama Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Original file line number Diff line number Diff 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+
2939runtime.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
95107runtime.command_alias(
You can’t perform that action at this time.
0 commit comments