From 432c02dabc7404c551c7be5b1c4ec0d59117d3d1 Mon Sep 17 00:00:00 2001
From: Angelos Katharopoulos <a_katharopoulos@apple.com>
Date: Tue, 19 Aug 2025 00:02:27 -0700
Subject: [PATCH] Typo in test

---
 python/tests/test_fast.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tests/test_fast.py b/python/tests/test_fast.py
index 518c26c70..5aabaf388 100644
--- a/python/tests/test_fast.py
+++ b/python/tests/test_fast.py
@@ -585,7 +585,7 @@ class TestFast(mlx_tests.MLXTestCase):
     def test_custom_kernel_basic(self):
         if mx.metal.is_available():
             source = """
-                uint elem thread_position_in_grid.x;
+                uint elem = thread_position_in_grid.x;
                 out1[elem] = a[elem];
             """
             custom_kernel = mx.fast.metal_kernel