pytorch/aten
Manuel Candales d6115439be [MPS] Add SDPA implentation (#131362)
This work is based off @malfet's #119200

Co-authored-by: Nikita Shulga <nikita.shulga@gmail.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/131362
Approved by: https://github.com/kimishpatel
2024-07-25 03:24:37 +00:00
..
conda
src [MPS] Add SDPA implentation (#131362) 2024-07-25 03:24:37 +00:00
tools [BE][CI] Get rid of duplicated code (#131406) 2024-07-23 04:01:13 +00:00
CMakeLists.txt