OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00
Code Issues Actions 52 Packages Projects Releases Wiki Activity

[MPS] Add mps keys to indices and values ops (#160223)

Browse Source 
enable indices and values on sparse mps

Pull Request resolved: https://github.com/pytorch/pytorch/pull/160223
Approved by: https://github.com/malfet
This commit is contained in:
Isalia20
2025-08-12 22:08:44 +00:00
committed by PyTorch MergeBot
parent 16d15445f8
commit 2e4e5ab4be
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
aten/src/ATen/native/native_functions.yaml
View File

@@ -7462,7 +7462,7 @@
- func: indices(Tensor(a) self) -> Tensor(a)
variants: method
dispatch:
SparseCPU, SparseCUDA, SparseMeta: indices_sparse
SparseCPU, SparseCUDA, SparseMPS, SparseMeta: indices_sparse
CompositeExplicitAutograd: indices_default
device_check: NoCheck
device_guard: False
@@ -7470,7 +7470,7 @@
- func: values(Tensor(a) self) -> Tensor(a)
variants: method
dispatch:
SparseCPU, SparseCUDA, SparseMeta: values_sparse
SparseCPU, SparseCUDA, SparseMPS, SparseMeta: values_sparse
SparseCsrCPU, SparseCsrCUDA, SparseCsrMeta: values_sparse_csr
NestedTensorCPU, NestedTensorHPU, NestedTensorCUDA: values_nested
CompositeExplicitAutograd: values_default