From 79b133cf02819ac6984ebe60326383a68c5ce4d8 Mon Sep 17 00:00:00 2001 From: Chen Lai Date: Fri, 3 Jan 2025 15:21:10 -0800 Subject: [PATCH] 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 --- examples/qualcomm/oss_scripts/llama3_2/TARGETS | 3 +++ 1 file changed, 3 insertions(+) diff --git a/examples/qualcomm/oss_scripts/llama3_2/TARGETS b/examples/qualcomm/oss_scripts/llama3_2/TARGETS index e35f7153bdd..42efd4d90b4 100644 --- a/examples/qualcomm/oss_scripts/llama3_2/TARGETS +++ b/examples/qualcomm/oss_scripts/llama3_2/TARGETS @@ -9,6 +9,9 @@ python_binary( name = "llama", srcs = ["llama.py"], main_function = "executorch.examples.qualcomm.oss_scripts.llama3_2.llama.main", + preload_deps = [ + "//executorch/extension/llm/custom_ops:model_sharding_py", + ], deps = [ "//executorch/examples/qualcomm/oss_scripts/llama2:static_llama", "//caffe2:torch",