pytorch

59d6e07a - fix forward_idx check (#59911)

fix forward_idx check (#59911) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/59911 Reviewed By: dzhulgakov Differential Revision: D29829020 Pulled By: albanD fbshipit-source-id: f685063061dab499368a272d6b94a44e89f9a143