8 Commits

Author	SHA1	Message	Date
Laurent Mazare	9dbaf958dc	Add an enum for scalar values. (#2909) ... * Add a scalar enum type. * Add a bit more to the scalar type. * Small tweak. * More scalar usage.	2025-04-18 22:13:38 +02:00
Anubhab Bandyopadhyay	54e7fc3c97	Lint fixes introduced with Rust 1.83 (#2646) ... * Fixes for lint errors introduced with Rust 1.83 * rustfmt * Fix more lints. --------- Co-authored-by: Laurent <laurent.mazare@gmail.com>	2024-11-28 23:00:21 +01:00
Laurent Mazare	0e2c8c17fb	UG metal integration. (#2580)	2024-10-27 15:20:37 +01:00
Laurent Mazare	af2104078f	Metal commands refactoring (#2489) ... * Split out the commands part of the metal device. * Make most fields private. * Move the allocator back. * Rework the encoder provider type.	2024-09-21 13:18:42 +02:00
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
Laurent Mazare	ddafc61055	Use RAII for terminating the encoding. (#2353)	2024-07-24 16:29:56 +02:00
Laurent Mazare	a925ae6bc6	Use a trait for the encoder provider (so that encoder can ultimately be reused). (#2352)	2024-07-24 09:27:30 +02:00
Laurent Mazare	c5fe4a7f89	Rework the buffer offset logic for metal kernels (#2028) ... * Move the metal kernels utils in a separate module. * Use the BufferOffset for unary ops. * Fix clippy lints. * Use the new BufferOffset. * Adapt the binary ops. * Affine. * More ops (powf, elu, cast).	2024-04-07 22:37:53 +02:00