3 Commits

Author	SHA1	Message	Date
Laurent Mazare	b6f7dfb682	CPU implementation for the custom RMS example. (#228) ... * CPU implementation for the custom RMS example. * Add the eps parameter.	2023-07-23 20:04:20 +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