I feel like in his prime it would be Zaha, but it's like Zaha is lacking motivation this season so hasn't bothered diving as much (of course he's still doing it, just not a lot of point when you're not overly invested), whereas Grealish still has that desire to fight for everything for his team...