fix a bug for size mismatch See merge request ADLR/megatron-lm!438
Attach a file by drag & drop or click to upload
