[Pytorch] fuse lm_head and cross_entropy#2609
Draft
Jianbing-D wants to merge 2 commits intoNVIDIA:mainfrom
Draft
[Pytorch] fuse lm_head and cross_entropy#2609Jianbing-D wants to merge 2 commits intoNVIDIA:mainfrom
Jianbing-D wants to merge 2 commits intoNVIDIA:mainfrom
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}