You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
11 lines
233 B
11 lines
233 B
1 month ago
|
import mlx.core as mx
|
||
|
|
from tinygrad.helpers import Timing
|
||
|
|
N = 4096
|
||
|
|
x = mx.random.normal((N,N))
|
||
|
|
w = mx.random.normal((N,N))
|
||
|
|
|
||
|
|
FLOPS = N*N*N*2
|
||
|
|
for i in range(10):
|
||
|
|
with Timing("", lambda x: f" {FLOPS/x:.2f} GFLOPS"):
|
||
|
|
mx.eval(x@w)
|