fix bf16 check when preprocessing data

fix bf16 check when preprocessing data #56