mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-08-21 08:14:00 +08:00
Vehicle Researcher
33d5cfc393
date: 2025-12-18T23:23:16 master commit: 3cdee7b54718ee14bd85befd6c5bad3d699c5479
11 lines
233 B
Python
11 lines
233 B
Python
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)
|