pytorch

b0814b63 - Reenable assert after test update

Reenable assert after test update Pull Request resolved: https://github.com/pytorch/pytorch/pull/78658 Approved by: https://github.com/ezyang, https://github.com/albanD