[FIX] Enable TORCH_CHECK again (#71971)
Summary:
From https://github.com/pytorch/pytorch/pull/71947, the `TORCH_CHECK` should be fixed now.
cc: albanD NSProgrammer
Pull Request resolved: https://github.com/pytorch/pytorch/pull/71971
Reviewed By: mrshenli
Differential Revision: D33845192
Pulled By: albanD
fbshipit-source-id: 020cbe1504ef6dd54f703d7bbc57c2cd22253363
(cherry picked from commit a154fdfb727f3762a8e1c9c71e48222ecdb3966e)
