def METHOD_NAME(
    self,
    tokens,
    lprobs,
    bsz: int,
    beam_size: int,
    step: int,
):
    return ngram_repeat_block_cuda.forward(
        tokens, lprobs, bsz, step, beam_size, self.no_repeat_ngram_size
    )