We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 03efc32 commit d0f3ccdCopy full SHA for d0f3ccd
exir/program/_program.py
@@ -1016,6 +1016,9 @@ def keep(op):
1016
torch.ops.aten.add.Tensor,
1017
torch.ops.aten.sub.Tensor,
1018
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,
1022
]:
1023
logging.warn(
1024
f"Op {op} was requested for preservation by partitioner. This request is ignored because it is in a blocklist."
0 commit comments