Dev 4bit embedding

[haowhsu-quic]

Summary

• 4bit embedding custom op
• trial run with story llama

Test plan

python examples/qualcomm/oss_scripts/llama/llama.py -s $DEVICE -m SM8650 -b build-android/ -a $ARTIFACT_DIR --ptq 16a4w --tokenizer_model PATH/TO/tokenizer.model --checkpoint PATH/TO/stories110M.pt --params PATH/TO/params.json --llama_model "stories110m" --tokenizer_bin PATH/TO/tokenizer.bin --prompt "Once upon" --model_mode kv --max_seq_len 128 --use_custom_embedding

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/8962

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 4 New Failures, 1 Cancelled Job

As of commit 8e200b7 with merge base e2201c5 ():

NEW FAILURES - The following jobs have failed:

• Check Labels / Check labels (gh)
  RuntimeError: Error checking labels: PR does not have required labels
• Lint / lintrunner / linux-job (gh)
  >>> Lint for examples/qualcomm/oss_scripts/llama/llama.py:
• pull / android / build-llm-demo / linux-job (gh)
  RuntimeError: Command docker exec -t daf82d14d57d6088d87b57c31fd6cbfcfe55607654e4972f5db4942bc9a29861 /exec failed with exit code 1
• pull / unittest-editable / macos / macos-job (gh)
  backends/xnnpack/test/ops/test_conv1d.py::TestConv1d::test_qs8_conv1d_batchnorm_seq

CANCELLED JOB - The following job was cancelled. Please retry:

• pull / test-static-llama-qnn-linux / linux-job (gh)
  ##[error]The operation was canceled.

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[github-actions]

This PR needs a release notes: label

If your changes are user facing and intended to be a part of release notes, please use a label starting with release notes:.

If not, please add the topic: not user facing label.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "topic: not user facing"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

[facebook-github-bot]

@billmguo has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.