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