drivers: ufs: synchronize upstream code

Signed-off-by: Yifeng Zhao <yifeng.zhao@rock-chips.com>
Change-Id: Ie8d0a0915e290b0e6d134a0d84653d63ae695914