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

[MTIA Aten Backend] Migrate max.dim_max / min.dim_min (#156568)

Browse Source 
# Context

See the first PR https://github.com/pytorch/pytorch/pull/153670

# This diff

 Migrate max.dim_max / min.dim_min to in-tree.

Differential Revision: [D77095185](https://our.internmc.facebook.com/intern/diff/D77095185/)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/156568
Approved by: https://github.com/malfet
ghstack dependencies: #156502, #156539, #156554
This commit is contained in:
anwang 2025-06-22 17:22:02 -07:00 committed by PyTorch MergeBot
parent dfdd636cfa
commit 31e1274597
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -3855,7 +3855,7 @@
precomputed:
- dim -> int dim
dispatch:
CPU, CUDA: max_out
CPU, CUDA, MTIA: max_out
MPS: max_out_mps
- func: max.names_dim(Tensor self, Dimname dim, bool keepdim=False) -> (Tensor values, Tensor indices)
@ -4044,7 +4044,7 @@
precomputed:
- dim -> int dim
dispatch:
CPU, CUDA: min_out
CPU, CUDA, MTIA: min_out
MPS: min_out_mps
- func: min.names_dim(Tensor self, Dimname dim, bool keepdim=False) -> (Tensor values, Tensor indices)