4 Commits

Author	SHA1	Message	Date
ivarflakstad	e90bcdcc7c	Metal: f16 and bf16 where_cond + benchmark (#1545) ... * Use cfg to seperate benchmark results based on features * Add metal where_cond for f16 and bf16. Add benchmark * Remove allow pragma * Avoid some unnecessary returns. * Improve benchmarks layout * Updated feature separated benchmarks --------- Co-authored-by: Laurent <laurent.mazare@gmail.com>	2024-01-12 11:18:11 +01:00
Laurent Mazare	89b5a06858	Use bindgen-cuda for the custom-kernel example. (#1536) ... * Use bindgen-cuda for the custom-kernel example. * Only depend on the kernels when cuda is enabled. * Skip rustfmt.	2024-01-07 17:18:46 +01:00
Laurent Mazare	e449ce53a2	Wrapping code to call the custom op. (#225) ... * Wrapping code to call the custom op. * Get the rms example to work. * Get around rustfmt failing in the CI. * Fix the rms computation.	2023-07-23 11:31:17 +01:00
Laurent Mazare	b8a10425ad	Kernel build example (#224) ... * Build example kernels. * Add some sample custom kernel. * Get the example kernel to compile. * Add some cuda code. * More cuda custom op. * More cuda custom ops.	2023-07-23 07:15:37 +01:00