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(
26
26
],
27
27
)
28
28
29
+ runtime.python_library(
30
+ name = "masking_utils",
31
+ srcs = [
32
+ "masking_utils.py",
33
+ ],
34
+ deps = [
35
+ "//caffe2:torch",
36
+ ],
37
+ )
38
+
29
39
runtime.python_library(
30
40
name = "decoder_constants",
31
41
srcs = [
@@ -39,6 +49,7 @@ runtime.python_library(
39
49
deps = [
40
50
":decoder_constants",
41
51
":decoder_utils",
52
+ ":masking_utils",
42
53
"//executorch/examples/models/llama:source_transformation",
43
54
"//caffe2:torch",
44
55
"//executorch/backends/qualcomm/partition:partition",
@@ -90,6 +101,7 @@ python_binary(
90
101
"//executorch/examples/qualcomm/oss_scripts/llama:range_setting_pt2e",
91
102
"fbsource//third-party/pypi/lm-eval:lm-eval",
92
103
],
104
+ keep_gpu_sections = True,
93
105
)
94
106
95
107
runtime.command_alias(
You can’t perform that action at this time.
0 commit comments