Anubhab Bandyopadhyay a2bcc227df Efficient implementation of Tensor::ones() for metal (#2512) ...

* WIP: hopefully better const impl

* with GPU

* More tests on

* Reverting primitive for

* Incorporating review changes - added check elem count check in kerner, using  for call strategy

* rustfmt ran

2024-10-01 19:11:59 +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

Efficient implementation of Tensor::ones() for metal (#2512)

2024-10-01 19:11:59 +02:00

indexing.metal

Support gather on bf16 for metal. (#2035)

2024-04-10 12:49:25 +02:00

lib.rs

Efficient implementation of Tensor::ones() for metal (#2512)

2024-10-01 19:11:59 +02:00

libMetalFlashAttention.metallib

Revert the bf16 gemm metal changes for now. (#2386)

2024-08-01 23:08:47 +02:00

mlx_gemm.metal

Integrate the MLX gemm kernels (#2468)

2024-09-11 16:56:48 +02: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

Add the layernorm specialized op. (#2212)

2024-05-24 15:58:01 +02:00

sort.metal

Add argsort. (#2132)

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

ternary.metal

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

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

tests.rs

Efficient implementation of Tensor::ones() for metal (#2512)

2024-10-01 19:11:59 +02:00

unary.metal

Fix for metal tanh. (#2475)

2024-09-13 07:08:36 +02:00

utils.rs

Metal commands refactoring (#2489)

2024-09-21 13:18:42 +02:00