From 284db26a120ed2bbd8d4055c5dc37e2dac07d95f Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 7 Oct 2025 12:13:01 +0800 Subject: [PATCH] cleanup --- extra/thunder/gemm.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/extra/thunder/gemm.py b/extra/thunder/gemm.py index ca69d0112a..61d3dae787 100644 --- a/extra/thunder/gemm.py +++ b/extra/thunder/gemm.py @@ -45,7 +45,6 @@ instantiate_matmul_custom(float32, float); from tinygrad import Device, Tensor, Context if __name__ == "__main__": - # TODO: why isn't this type inferred? device = Device["METAL"] lib = device.compiler.compile(gemm) prg = device.runtime("matmul_custom_float32", lib) @@ -70,7 +69,6 @@ if __name__ == "__main__": with Context(DEBUG=2): ref = (a@b).realize() - val = (ref-c).mean() - print(val.item()) + print((ref-c).mean().item())