code: def bench_add_sentence_boundary_token_ids(benchmark):
      device = torch.device("cuda")
      # shape: (32, 50)
      tensor = torch.tensor([[3] * 50] * 32, device=device)
      # shape: (32, 50)
      mask = torch.tensor(
          [[True] * 50, [True] * 30 + [False] * 20] * 16, device=device
      )
      begin_token = 1
      end_token = 2
      benchmark(
          util.add_sentence_boundary_token_ids,
          tensor,
          mask,
          begin_token,
          end_token,
      )
  
imports: import torch
  from allennlp.nn import util
  from allennlp.common.testing import requires_gpu
label_desc: Use image node_id 1 for calling a global function with example usage: bench_add_sentence_boundary_token_ids(benchmark) without return types