Changing return of relay.Function or IRModule?

[mbrookhart]

@jmatai1

        if call.op.name =="add":
            # save this call for return!
            self.return_values.append(call)
        super().visit_call(call)

        post = super().visit_call(call)
        if post.op.name =="add":
            # save this call for return!
            self.return_values.append(post)
        return post

You were storing the add pre-mutation.