megatron.model.module.fp32_to_float16

megatron.model.module.fp32_to_float16(val, float16_convertor)

Convert fp32 val to fp16/bf16