zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-06 17:31:13 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
ec0d5db67b
mlx/python/tests
History
Awni Hannun ec0d5db67b
[CUDA] Switch to CUDA graphs (#2317) 
* cuda graph prototype

fix signal bug + start to add dependencies

capture more

capture more ops

remaining ops

fix reduce and rope deps

add concurrent context

try update, but not working

cosistent topology order

use node api

use node api directly to reduce overhead

fix bug

use kernels in unary

cache graph

format

fix synchronization

format

* comment
2025-07-02 15:59:13 -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 use fp32 for testing, add more complex ops (#2322) 2025-07-01 07:30:00 -07:00
mlx_distributed_tests.py Distributed layers (#1270) 2025-03-21 13:52:17 -07:00
mlx_tests.py use fp32 for testing, add more complex ops (#2322) 2025-07-01 07:30:00 -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 Split broadcast so it is always fused in compile (#2318) 2025-06-26 22:08:18 -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 [CUDA] Switch to CUDA graphs (#2317) 2025-07-02 15:59:13 -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 allow parameters to be deleted (#2325) 2025-07-01 21:27:23 -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