nv cross entropy allow fp16 and none reduction#2446
Draft
liqiangxl wants to merge 2 commits intoLightning-AI:mainfrom
Draft
nv cross entropy allow fp16 and none reduction#2446liqiangxl wants to merge 2 commits intoLightning-AI:mainfrom
liqiangxl wants to merge 2 commits intoLightning-AI:mainfrom
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}