jhavukainen 8b9a3f3cea Align mlx::core::max op nan propagation with NumPy (#2339) ... * Make max op NaN propagation rules align with numpy * Adding benchmarks and testing for max op nanpropagation * Pre-commit formatting * Fix max complex64 nan propagation and add test * Improve the cpp unittest * Only check nans on non-integral types in simd_reduce_impl. * Cleanup using namespace alias * Add cpu Max nanpropagation. Fix a small fib in cpu max dispatch data types for int8/int16. * Make the max nanpropagation test more meaningful for integer types * Remove tuple unpacking syntax to comply with earlier python versions. Add cuda skip to nanpropagation tests, fix cuda implementation in a separate PR.		2025-07-09 11:26:27 -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	Align mlx::core::max op nan propagation with NumPy (#2339)	2025-07-09 11:26:27 -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	MoE backward improvements (#2335)	2025-07-07 17:59:53 -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	MoE backward improvements (#2335)	2025-07-07 17:59:53 -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	Align mlx::core::max op nan propagation with NumPy (#2339)	2025-07-09 11:26:27 -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