benchmark(torch.compile(opt_sample_data), input_samples, labels)