No, you do not wait on IRS to tell you. First, do the Form 1040-NR and figure out if you owe IRS or get a refund by following Form 1040-X instructions. Then if you will be getting more, send the amendment and wait on IRS to send you the difference. If you owe them, send the difference back to them with the amendment.
