why KimiMoEGate cast bf16 router‘s weight to fp32,instead of using torch.mm(..., out_dtype=float)

#186
by zz16822 - opened

why KimiMoEGate cast bf16 weight to fp32,instead of using torch.mm(..., out_dtype=float)

Sign up or log in to comment