eed21eeff2
* fix cast from bfloat16 * fix cast to bfloat16 test as well * clean up comments * lint * add comment Co-authored-by: Margaret Qian <mqian@octoml.ai>