Skip to content

Commit 79b133c

Browse files
cccclaifacebook-github-bot
authored andcommitted
Add custom ops deps
Summary: Add dependency in buck for model sharding op ``` buck run mode/dev-nosan //executorch/examples/qualcomm/oss_scripts/llama3_2:llama_qnn -- --compile_only --ptq 16a4w --checkpoint /home/chenlai/local//consolidated.00.pth --params /home/chenlai/local//params.json --tokenizer_model /home/chenlai/local//tokenizer.model --prompt "Once" -m SM8650 --model_size 1B --model_mode hybrid --build_folder /home/chenlai/local 2>&1 | tee static_llama.log ``` Differential Revision: D67814096
1 parent c001634 commit 79b133c

File tree

1 file changed

+3
-0
lines changed
  • examples/qualcomm/oss_scripts/llama3_2

1 file changed

+3
-0
lines changed

examples/qualcomm/oss_scripts/llama3_2/TARGETS

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,9 @@ python_binary(
99
name = "llama",
1010
srcs = ["llama.py"],
1111
main_function = "executorch.examples.qualcomm.oss_scripts.llama3_2.llama.main",
12+
preload_deps = [
13+
"//executorch/extension/llm/custom_ops:model_sharding_py",
14+
],
1215
deps = [
1316
"//executorch/examples/qualcomm/oss_scripts/llama2:static_llama",
1417
"//caffe2:torch",

0 commit comments

Comments
 (0)