torch.testing.assert_close(compiled_output, eager_output)
torch.testing.assert_close(repeated_output, eager_output)

max_abs_diff = (eager_output - compiled_output).abs().max().item()
print(f"torch={torch.__version__}")
print(f"input_shape={tuple(sample.shape)}")
print(f"output_shape={tuple(compiled_output.shape)}")
print(f"max_abs_diff={max_abs_diff:.8f}")
print(f"compiled_matches_eager={torch.allclose(compiled_output, eager_output)}")
print(f"repeat_matches_eager={torch.allclose(repeated_output, eager_output)}")