fix mm not correctly report TORCH_CHECK failure issue (#61394)
Summary:
fixes https://github.com/pytorch/pytorch/issues/61291.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61394
Reviewed By: zhouzhuojie, seemethere
Differential Revision: D29614208
Pulled By: walterddr
fbshipit-source-id: f49a15dde708e30b06059b47fae1cda7c2c3571c
Author
Rong Rong (AI Infra)
