Skip to content

Commit 723c0f3

Browse files
Add buck dependencies for lowbit embedding and linear flow (#17802)
Summary: Now that they are enabled internally expose the ops deps in buck Differential Revision: D94729377
1 parent edaf3ef commit 723c0f3

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

  • extension/llm/export

extension/llm/export/BUCK

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,8 @@ fbcode_target(_kind = runtime.python_binary,
6969
"//executorch/extension/llm/custom_ops:model_sharding_py",
7070
"//executorch/extension/llm/custom_ops:custom_ops_aot_lib",
7171
"//executorch/kernels/quantized:aot_lib",
72+
"//pytorch/ao/torchao/csrc/cpu/shared_kernels/embedding_xbit:op_embedding_xbit_aten",
73+
"//pytorch/ao/torchao/csrc/cpu/shared_kernels/linear_8bit_act_xbit_weight:op_linear_8bit_act_xbit_weight_aten",
7274
],
7375
deps = [
7476
"fbsource//third-party/pypi/hydra-core:hydra-core",

0 commit comments

Comments
 (0)