frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[MPS] Add boilerplate sparse code support (#157238)

Browse Source 
This PR makes minimal changes to support sparse tensors on MPS. In the followup PRs I'll start adding different operations slowly so we can fix the issue of
https://github.com/pytorch/pytorch/issues/129842
which is highly requested(I assume because of whisper using sparse tensors)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/157238
Approved by: https://github.com/malfet
This commit is contained in:
Isalia20
2025-06-30 01:53:45 +00:00
committed by PyTorch MergeBot
parent 771be85704
commit a1282b1823
9 changed files with 97 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torchgen/model.py
View File

@ -288,6 +288,8 @@ dispatch_keys = [
DispatchKey.SparseCsrXPU,
DispatchKey.SparseCUDA,
DispatchKey.SparseCsrCUDA,
DispatchKey.SparseMPS,
DispatchKey.SparseCsrMPS,
DispatchKey.QuantizedCPU,
DispatchKey.QuantizedCUDA,
DispatchKey.CompositeImplicitAutograd,