Laurent Mazare e27b4700ad Indexing with max-value results in zero/no-op. (#2940) ...

* Indexing with max-value results in zero/no-op.

* Add some testing.

* Also adapt the metal kernels.

* Another test.

* Fix.

2025-05-03 11:36:31 +02:00

..

affine.metal

Bugfix for the metal elu kernel. (#2490)

2024-09-21 15:03:19 +02:00

binary.metal

update dtypes checks for several metal operations (#2010)

2024-04-04 18:39:06 +02:00

cast.metal

add support for casting between all datatypes (#1860)

2024-03-17 20:55:11 +01:00

conv.metal

Add a metal kernel for col2im1d. (#2214)

2024-05-25 11:03:23 +02:00

fill.metal

Add an enum for scalar values. (#2909)

2025-04-18 22:13:38 +02:00

indexing.metal

Indexing with max-value results in zero/no-op. (#2940)

2025-05-03 11:36:31 +02:00

lib.rs

Support for "unbatched" rope. (#2926)

2025-04-27 15:12:02 +02:00

mlx_gemm.metal

Integrate the MLX gemm kernels (#2468)

2024-09-11 16:56:48 +02:00

mlx_gemm.rs

Add the MLX merge sort kernels (#2751)

2025-01-28 14:09:43 +01:00

mlx_sort.metal

Add the MLX merge sort kernels (#2751)

2025-01-28 14:09:43 +01:00

quantized.metal

Add argsort. (#2132)

2024-04-27 20:17:35 +02:00

random.metal

Metal random-generation bug fixes (#1811)

2024-03-08 16:11:50 +01:00

reduce.metal

Support for "unbatched" rope. (#2926)

2025-04-27 15:12:02 +02:00

scaled_dot_product_attention.metal

Add a few metal gather ops. (#2740)

2025-01-25 23:31:03 +01:00

sort.metal

Add argsort. (#2132)

2024-04-27 20:17:35 +02:00

sort.rs

Add the MLX merge sort kernels (#2751)

2025-01-28 14:09:43 +01:00

ternary.metal

Add some missing where-cond kernels for metal. (#2203)

2024-05-22 09:44:52 +02:00

tests.rs

Support for "unbatched" rope. (#2926)

2025-04-27 15:12:02 +02:00

unary.metal

Add the const-set op. (#2910)

2025-04-19 10:07:02 +02:00

utils.metal

Metal: Improved reduce and softmax (#1819)

2025-02-08 07:27:01 +01:00

utils.rs

Add an enum for scalar values. (#2909)

2025-04-18 22:13:38 +02:00