Pre-commit formatting

2025-12-16 01:49:05 +08:00 · 2025-07-06 14:27:40 -07:00
parent af74818528
commit 5b089dc5da
4 changed files with 11 additions and 11 deletions
--- a/benchmarks/python/single_ops.py
+++ b/benchmarks/python/single_ops.py
@@ -50,12 +50,14 @@ def time_maximum():
    mx.eval(a, b)
    time_fn(mx.maximum, a, b)

+
 def time_max():
    a = mx.random.uniform(shape=(32, 1024, 1024))
-    a[1,1] = mx.nan
+    a[1, 1] = mx.nan
    mx.eval(a)
    time_fn(mx.max, a, 0)

+
 def time_negative():
    a = mx.random.uniform(shape=(10000, 1000))
    mx.eval(a)