zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 01:41:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
209404239b
mlx/python/tests
History
Angelos Katharopoulos 209404239b
Fix the accelerate dispatch for the power op (#70) 
- The exponent and base were swapped because accelerate is using
  exponent-base instead of base-exponent
- Fix also the test for binary ops as it was testing op(x, x) which
  couldn't catch ordering errors like that
2023-12-08 10:58:03 -08:00
..
mlx_tests.py copyright + ack 2023-11-30 11:12:53 -08:00
test_array.py Add CircleCI configuration (#4) 2023-12-04 16:04:11 -08:00
test_autograd.py copyright + ack 2023-11-30 11:12:53 -08:00
test_bf16.py copyright + ack 2023-11-30 11:12:53 -08:00
test_blas.py Fix gemv broadcasting bug (#6) 2023-12-05 14:15:43 -08:00
test_conv.py copyright + ack 2023-11-30 11:12:53 -08:00
test_device.py copyright + ack 2023-11-30 11:12:53 -08:00
test_eval.py copyright + ack 2023-11-30 11:12:53 -08:00
test_fft.py copyright + ack 2023-11-30 11:12:53 -08:00
test_load.py copyright + ack 2023-11-30 11:12:53 -08:00
test_nn.py copyright + ack 2023-11-30 11:12:53 -08:00
test_ops.py Fix the accelerate dispatch for the power op (#70) 2023-12-08 10:58:03 -08:00
test_optimizers.py copyright + ack 2023-11-30 11:12:53 -08:00
test_random.py copyright + ack 2023-11-30 11:12:53 -08:00
test_reduce.py copyright + ack 2023-11-30 11:12:53 -08:00
test_tree.py copyright + ack 2023-11-30 11:12:53 -08:00
test_vmap.py copyright + ack 2023-11-30 11:12:53 -08:00