refactor: use perf_counter for accurate benchmarking (#2940)

This commit is contained in:
Satyam singh
2025-12-21 19:37:00 +05:30
committed by GitHub
parent 7652f1c152
commit e6de81c963
+2 -2
View File
@@ -30,13 +30,13 @@ def loss_fn(w):
grad_fn = mx.grad(loss_fn)
tic = time.time()
tic = time.perf_counter()
for _ in range(num_iters):
grad = grad_fn(w)
w = w - lr * grad
mx.eval(w)
toc = time.time()
toc = time.perf_counter()
loss = loss_fn(w)
final_preds = (X @ w) > 0