angelayi aff9c1eec5 [aoti][mps] Add fused_rms and sdpa_mps fallback ops (#156844) ...

Needed for llama3.1

Pull Request resolved: https://github.com/pytorch/pytorch/pull/156844
Approved by: https://github.com/desertfire
ghstack dependencies: #156843

2025-06-26 21:03:05 +00:00

..

__init__.py

[AOTI] Refine the C shim autogen mechanism (#125589)

2024-05-09 02:48:16 +00:00

fallback_ops.py

[aoti][mps] Add fused_rms and sdpa_mps fallback ops (#156844)

2025-06-26 21:03:05 +00:00