From 01b055abe3669f036a60da6158ab778b0700baaf Mon Sep 17 00:00:00 2001
From: Jack Zhang <jackzhxng@meta.com>
Date: Mon, 28 Oct 2024 17:31:53 +0000
Subject: [PATCH] Make masked_scatter core aten (#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: https://github.com/pytorch/pytorch/pull/137949
Approved by: https://github.com/larryliu0820
---
 aten/src/ATen/native/native_functions.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/aten/src/ATen/native/native_functions.yaml b/aten/src/ATen/native/native_functions.yaml
index 3625cd87124..00b21cc25cc 100644
--- a/aten/src/ATen/native/native_functions.yaml
+++ b/aten/src/ATen/native/native_functions.yaml
@@ -8015,6 +8015,7 @@
   variants: function, method
   dispatch:
     CompositeExplicitAutograd: masked_scatter
+  tags: core
 
 - func: masked_scatter_backward(Tensor grad_output, Tensor mask, SymInt[] sizes) -> Tensor
   dispatch: