mirror of https://github.com/huggingface/candle.git synced 2025-06-15 18:28:24 +00:00

Files

Laurent Mazare 5635650d38 Integrate the MLX gemm kernels (#2468 )

* Include the MLX gemm kernels.

* Clippy lints.

* Export the gemm_f32 kernel.

* Add the f16/bf16 variants.

* Add the initial dispatch code.

* More plugging of the mlx kernels.

* Add a currently broken test.

* Tweaks.

* Bugfix + get the tests to pass.

* Enable the gemm bf16 tests.

* Add some randomized tests.

* Update candle-metal-kernels/src/lib.rs

Co-authored-by: ivarflakstad <69173633+ivarflakstad@users.noreply.github.com>

* More fixes.

* More clippy fixes.

---------

Co-authored-by: ivarflakstad <69173633+ivarflakstad@users.noreply.github.com>

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

src

Integrate the MLX gemm kernels (#2468 )

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

tmp

Starting to fix some tests.

2023-11-30 11:30:31 +01:00

Cargo.toml

Integrate the MLX gemm kernels (#2468 )

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

README.md

Adding the actual backend

2023-11-20 14:12:56 +01:00

README.md

candle-metal-kernels

This crate contains Metal kernels used from candle.