Diaz surprised co-star with her jab at tech billionaire’s friendship with Trump