Replace torch.jit.script with torch.compile in get_masked_input_and_mask to fix benchmark underreporting

Replace torch.jit.script with torch.compile in get_masked_input_and_mask to fix benchmark underreporting #20420