Skip to content

Commit 129d9a3

Browse files
billmguofacebook-github-bot
authored andcommitted
support more model lowering (#8967)
Summary: Pull Request resolved: #8967 Reviewed By: YIWENX14 Differential Revision: D70429474
1 parent bdafb22 commit 129d9a3

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

exir/program/_program.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1016,6 +1016,9 @@ def keep(op):
10161016
torch.ops.aten.add.Tensor,
10171017
torch.ops.aten.sub.Tensor,
10181018
torch.ops.aten.div.Tensor,
1019+
torch.ops.aten.item.default,
1020+
torch.ops.aten._local_scalar_dense.default,
1021+
torch.ops.aten.unbind.int,
10191022
]:
10201023
logging.warn(
10211024
f"Op {op} was requested for preservation by partitioner. This request is ignored because it is in a blocklist."

0 commit comments

Comments
 (0)