SIGN IN SIGN UP

Move llm custom ops to extension (#4467)

Summary:
Move sdpa from examples/models/llama2/custom_ops to extension/llm/custom_ops.

Consolidate llm custom ops into one place - later, other custom ops (eg. pad, reshape) can be placed in the same place.

Pull Request resolved: https://github.com/pytorch/executorch/pull/4467

Test Plan:
Imported from GitHub, without a `Test Plan:` line.

Build executorch, export and run stories with xnnpack and quant in oss.

Reviewed By: helunwencser

Differential Revision: D60469989

Pulled By: lucylq

fbshipit-source-id: e2950f13976c422fc51be8204a44f652faf2f9a9
L
lucylq committed
6cd7f380e31a1f53f4448ea2143e93efb8a9383c
Parent: 227b49d
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> on 7/31/2024, 1:12:32 AM