scsi: ufs: make HCE polling more compact to improve initialization latency