zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Typo in test

Browse Source
This commit is contained in:
Angelos Katharopoulos
2025-08-19 00:02:27 -07:00
parent fa555c536a
commit 432c02dabc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_fast.py
View File

@@ -585,7 +585,7 @@ class TestFast(mlx_tests.MLXTestCase):
def test_custom_kernel_basic(self): def test_custom_kernel_basic(self):
if mx.metal.is_available(): if mx.metal.is_available():
source = """ source = """
uint elem thread_position_in_grid.x; uint elem = thread_position_in_grid.x;
out1[elem] = a[elem]; out1[elem] = a[elem];
""" """
custom_kernel = mx.fast.metal_kernel custom_kernel = mx.fast.metal_kernel