zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-03 15:51:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix benchmark example (#11)

Browse Source
This commit is contained in:
Yingbo Ma 2023-12-06 10:17:16 -05:00 committed by GitHub
parent 8c96b9a890
commit 36b245b287
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
benchmarks/python/batch_matmul_bench.py
View File

@ -30,7 +30,7 @@ def time_batch_matmul():
time_fn(batch_vjp_second)
def time_unbatch_matmul(key):
def time_unbatch_matmul():
mx.random.seed(3)
a = mx.random.uniform(shape=(B * T, D))
b = mx.random.uniform(shape=(D, D))