-	if (!ufshpb_is_support_chunk(transfer_len))
-		return;
+	if (!ufshpb_is_support_chunk(hpb, transfer_len) &&
+	    (ufshpb_is_legacy(hba) && (transfer_len != 
HPB_LEGACY_CHUNK_HIGH)))
+		return 0;

This is looks awkward, can we put the checks in 
ufshpb_is_support_chunk()?

Thanks,

Can Guo.