How to register a custom operator with multiple in-place operations in PyTorch?

1

I’m trying to register an operator that contains multiple in-place operations, defined as:
- func: custom_demo_add_(Tensor(a!) input1, Tensor(b!) input2) → (Tensor(a!), Tensor(b!))
But I’m getting an error: ‘If you have a schema with mutable positional args, we expect them to not be returned.’ What should I do?

2

Perhaps you should not return them? Maybe like this → - func: custom_demo_add_(Tensor(a!) input1, Tensor(b!) input2) → ()