drm/msm: Use for_each_*_in_state