Laurent Mazare a2e925462c Add the scatter in place ops. (#2923) ...

* Add the scatter_set op.

* Metal op.

* Cuda version.

* Merge the checks.

* Add the actual ops.

2025-04-26 07:36:49 +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

Add the scatter op. (#2921)

2025-04-25 21:46:58 +02:00

lib.rs

Add the scatter in place ops. (#2923)

2025-04-26 07:36:49 +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

Metal: Improved reduce and softmax (#1819)

2025-02-08 07:27:01 +01: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

Add the scatter op. (#2921)

2025-04-25 21:46:58 +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