Update wait mechanism and add different bit width of input data