SIGN IN SIGN UP

apply simple sdpa to coreml/mps backend (#3660)

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

coreml and mps doesn't support sdpa at the moment, use simple sdpa to have a simpler decomposition. Observer 1.5x faster on emulator.

Reviewed By: shoumikhin, kirklandsign

Differential Revision: D57476985

fbshipit-source-id: 2dbcad1a6e8b744e0a95d60fcd740369d665eab0
C
Chen Lai committed
a32e729fa30fccd0de4514fbd70cba8d9b94f5aa
Parent: 39834b1
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> on 5/20/2024, 6:16:22 PM