Musnad Abu Ya'la al-Mawsili 2:18

Ubayd Allah ibn Umar narrated to us, Yazid ibn Zuray' narrated to us, Sa'id ibn Abi Arubah narrated to us, from Qatadah, from Sa'id ibn al-Musayyib, from Ibn Umar, from Umar, from the Prophet (may Allah bless him and grant him peace), who said: "Indeed, the deceased is punished in his grave for what is wailed over him, or for what is wept over him."