I think it would depend on how he is with you outside of using his phone? If everything else is ok, then I’d just leave him alone if you see him on his phone. If this is a symptom of a wider problem, it might be worth having a conversation or consider what you want to do, if he doesn’t want to change.