hi,
maybe it’s because of retaining input for computing grad, if you don’t need to call backward, you can use
@torch.no_grad()
def _add(self, *inputs):
        l, r = inputs
        return torch.add(l, r)

i’m not sure about it.