fba10af9db5edd61f78ccdb9d115c4eafbcc561d,fairseq/search.py,BeamSearch,step,#BeamSearch#Any#Any#Any#,59

Before Change

            ),
            out=(self.scores_buf, self.indices_buf),
        )
        torch.div(self.indices_buf, vocab_size, out=self.beams_buf)
        self.indices_buf.fmod_(vocab_size)
        return self.scores_buf, self.indices_buf, self.beams_buf

After Change

        )
        self.scores_buf = top_prediction[0]
        self.indices_buf = top_prediction[1]
        self.beams_buf = torch.div(self.indices_buf, vocab_size)
        self.indices_buf.fmod_(vocab_size)
        return self.scores_buf, self.indices_buf, self.beams_buf

Instances