Skip to content

Commit d0f3ccd

Browse files
billmguofacebook-github-bot
authored andcommitted
support more model lowering (#8967)
Summary: Pull Request resolved: #8967 Differential Revision: D70429474
1 parent 03efc32 commit d0f3ccd

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)