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

Make masked_scatter core aten (#137949)

Browse Source 
Summary: Making `masked_scatter` core aten since it is hard to decompose and we now have a portable kernel for it

Test Plan: N/A

Differential Revision: D64368725

Pull Request resolved: https://github.com/pytorch/pytorch/pull/137949
Approved by: https://github.com/larryliu0820
This commit is contained in:
Jack Zhang 2024-10-28 17:31:53 +00:00 committed by PyTorch MergeBot
parent bca696ae81
commit 01b055abe3
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -8015,6 +8015,7 @@
variants: function, method variants: function, method
dispatch: dispatch:
CompositeExplicitAutograd: masked_scatter CompositeExplicitAutograd: masked_scatter
tags: core
- func: masked_scatter_backward(Tensor grad_output, Tensor mask, SymInt[] sizes) -> Tensor - func: masked_scatter_backward(Tensor grad_output, Tensor mask, SymInt[] sizes) -> Tensor
dispatch: dispatch: