mlx/python/tests
2025-06-23 05:49:49 -07:00
..
__main__.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
cuda_skip.py Fixes for transpositions and expands 2025-06-23 05:49:49 -07:00
mlx_distributed_tests.py Distributed layers (#1270) 2025-03-21 13:52:17 -07:00
mlx_tests.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
mpi_test_distributed.py Fix the test and add custom min/max reductions for uncommon MPI types (#2060) 2025-04-10 17:01:17 -07:00
ring_test_distributed.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_array.py [CUDA] Fix back-end bugs and enable corresponding tests (#2296) 2025-06-16 08:45:40 -07:00
test_autograd.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_bf16.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_blas.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_compile.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_constants.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_conv_transpose.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_conv.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_device.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_double.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_einsum.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_eval.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_export_import.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_fast_sdpa.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_fast.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_fft.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_graph.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_init.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_linalg.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_load.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_losses.py Cuda bug fixes 2 (#2298) 2025-06-16 13:14:46 -07:00
test_memory.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_nn.py Fix update_modules() when providing a subset (#2308) 2025-06-20 17:19:46 -07:00
test_ops.py [CUDA] Fix back-end bugs and enable corresponding tests (#2296) 2025-06-16 08:45:40 -07:00
test_optimizers.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_quantized.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_random.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_reduce.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_tree.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_upsample.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00
test_vmap.py add python testing for cuda with ability to skip list of tests (#2295) 2025-06-15 10:56:48 -07:00