Ghafir 40:46 mentions only punishment after resurrection and no earlier stage.

Recording Angels, the Angel of Death, and Questioning and Punishment in the Grave