import torch
import intel_extension_for_pytorch as ipex
import time

# Model simplu
model = torch.nn.Conv2d(3, 64, 3)
input_data = torch.randn(1, 3, 224, 224)

# Fără IPEX
start = time.time()
output = model(input_data)
print("Timp fără IPEX:", time.time() - start)

# Cu IPEX
model, _ = ipex.optimize(model)
start = time.time()
output = model(input_data)
print("Timp cu IPEX:", time.time() - start)