CCYeh b3825ac149 Add Masked Scatter (#2663) ...

Co-authored-by: Awni Hannun <awni@apple.com>
Co-authored-by: Angelos Katharopoulos <katharas@gmail.com>
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>

2025-11-19 14:53:32 -08: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

Add Masked Scatter (#2663)

2025-11-19 14:53:32 -08:00

mlx_distributed_tests.py

Nccl reduce scatter, all gather (#2727)

2025-11-05 08:21:11 -08:00

mlx_tests.py

Detect cache thrashing in LRUCache (#2600)

2025-09-19 09:12:14 +09:00

mpi_test_distributed.py

Nccl reduce scatter, all gather (#2727)

2025-11-05 08:21:11 -08:00

nccl_test_distributed.py

Nccl reduce scatter, all gather (#2727)

2025-11-05 08:21:11 -08:00

ring_test_distributed.py

Nccl reduce scatter, all gather (#2727)

2025-11-05 08:21:11 -08:00

test_array.py

Add Masked Scatter (#2663)

2025-11-19 14:53:32 -08:00

test_autograd.py

cuda fix flaky test (#2646)

2025-10-02 15:40:04 -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

Fix addmm with empty matrices and beta != 1.0 (#2715)

2025-11-03 14:16:15 -08:00

test_compile.py

Shapeless support for zeros/ones_like (#2726)

2025-11-06 19:12:20 -08: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

Fix overflow in large filter small channels (#2520)

2025-08-20 08:03:29 -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

Fix exporting with constants (#2769)

2025-11-14 12:52:08 -08:00

test_fast_sdpa.py

[CUDA] cuDNN backward attention (#2762)

2025-11-19 08:13:50 +09:00

test_fast.py

more accurate rope fallback (#2792)

2025-11-19 06:07:21 -08: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 load with path tests (#2543)

2025-08-26 14:24:47 -07:00

test_losses.py

fix cross entropy axis param (#2641)

2025-10-01 16:49:55 -07:00

test_memory.py

fix memory count bug (#2717)

2025-10-30 14:27:15 -07:00

test_nn.py

Add mode parameter for quantization (#2499)

2025-08-28 06:45:26 -07:00

test_ops.py

Add Masked Scatter (#2663)

2025-11-19 14:53:32 -08:00

test_optimizers.py

Adding support for the Muon Optimizer (#1914)

2025-07-18 12:25:28 -07:00

test_quantized.py

Add quantize/dequantize for mxfp8 and nvfp4 (#2688)

2025-10-28 16:23:12 -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

fix complex reduce + nan propagation in min and max (#2377)

2025-07-15 18:19:47 -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 Masked Scatter (#2663)

2025-11-19 14:53:32 -08:00