From 129d9a3419d3829bb574f4d499163c6228887857 Mon Sep 17 00:00:00 2001
From: Min Guo <minguo@meta.com>
Date: Thu, 6 Mar 2025 13:51:52 -0800
Subject: [PATCH] support more model lowering (#8967)

Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/8967

Reviewed By: YIWENX14

Differential Revision: D70429474
---
 exir/program/_program.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/exir/program/_program.py b/exir/program/_program.py
index 8295907d090..ed9dace34d1 100644
--- a/exir/program/_program.py
+++ b/exir/program/_program.py
@@ -1016,6 +1016,9 @@ def keep(op):
             torch.ops.aten.add.Tensor,
             torch.ops.aten.sub.Tensor,
             torch.ops.aten.div.Tensor,
+            torch.ops.aten.item.default,
+            torch.ops.aten._local_scalar_dense.default,
+            torch.ops.aten.unbind.int,
         ]:
             logging.warn(
                 f"Op {op} was requested for preservation by partitioner.  This request is ignored because it is in a blocklist."