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

Register MTIA kernel for all_all_out (#164293)

Browse Source 
Reviewed By: srsuryadev

Differential Revision: D83517879

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164293
Approved by: https://github.com/Skylion007, https://github.com/malfet
This commit is contained in:
Tristan Trouwen
2025-10-01 09:05:08 +00:00
committed by PyTorch MergeBot
parent 17ab99463a
commit 9fd53a2bdc
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -10256,6 +10256,7 @@
structured: True
dispatch:
CPU, CUDA: all_all_out
MTIA: all_all_out_mtia
MPS: all_all_out_mps
- func: any(Tensor self) -> Tensor