AMGStep: override `apply`

The generic version doesn't know about the `residual_` member.

It would be better if the generic version would call
`setProblem(x, rhs)`, but this variant of `setProblem` doesn't exist.