Input: ambakmi - fix timeout handling in amba_kmi_write()