zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-21 20:46:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

spelling: perturbations

Browse Source 
Signed-off-by: Josh Soref <2119212+jsoref@users.noreply.github.com>
This commit is contained in:
Josh Soref 2024-01-01 22:43:22 -05:00
parent f890871ed7
commit 039ee91238
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/tests/test_blas.py
View File

@ -53,10 +53,10 @@ class TestBlas(mlx_tests.MLXTestCase):
for dtype in self.dtypes: for dtype in self.dtypes:
np_dtype = getattr(np, dtype) np_dtype = getattr(np, dtype)
base_shapes = [4, 8, 16, 32, 64, 128] base_shapes = [4, 8, 16, 32, 64, 128]
pertubations = [-2, -1, 0, 1, 2] perturbations = [-2, -1, 0, 1, 2]
for dim in base_shapes: for dim in base_shapes:
for p in pertubations: for p in perturbations:
shape_a = (dim + p, dim + p) shape_a = (dim + p, dim + p)
shape_b = (dim + p, dim + p) shape_b = (dim + p, dim + p)
self.__gemm_test(shape_a, shape_b, np_dtype) self.__gemm_test(shape_a, shape_b, np_dtype)