Skip to content

Commit 01b055a

Browse files
jackzhxngpytorchmergebot
authored andcommitted
Make masked_scatter core aten (pytorch#137949)
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: pytorch#137949 Approved by: https://github.com/larryliu0820
1 parent bca696a commit 01b055a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

aten/src/ATen/native/native_functions.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8015,6 +8015,7 @@
80158015
variants: function, method
80168016
dispatch:
80178017
CompositeExplicitAutograd: masked_scatter
8018+
tags: core
80188019

80198020
- func: masked_scatter_backward(Tensor grad_output, Tensor mask, SymInt[] sizes) -> Tensor
80208021
dispatch:

0 commit comments

Comments
 (0)