Modify assert order to correct the error message when nan appears in multinomial on cuda (#53288)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/53288
Modify assert order to correct the error message when nan appears in multinomial on cuda
Test Plan: unittest
Reviewed By: ngimel
Differential Revision: D26824353
fbshipit-source-id: af6195e7c36fd51b3fc90df558ad6fac41288142