Awni Hannun cad5c0241c [CUDA] synch properly waits for all tasks to finish and clear (#2303) ...

* cuda synch properly waits for all tasks to finish and clear

* fix copy

2025-06-17 12:03:25 -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

[CUDA] synch properly waits for all tasks to finish and clear (#2303)

2025-06-17 12:03:25 -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

add python testing for cuda with ability to skip list of tests (#2295)

2025-06-15 10:56:48 -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