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."