Split SDPA + KV cache operator into SDPA operator and KV cache update operator + Add `RemoveAsserts` pass and apply it during LlaMa export
Differential Revision: D68922404 Pull Request resolved: https://github.com/pytorch/executorch/pull/8075
S
Sicheng Stephen Jia committed
ca7cdc82332ee8e6ed372a55c8093e3cc8f148aa
Parent: e8ee36c
Committed by GitHub <noreply@github.com>
on 1/31/2025, 8:55:23 PM