Laurent Mazare ce9fbc3682 Optimize the cat operation on contiguous tensors (#1855) ... * Add a specialized kernel for copy2d. * Move the cat operations. * Avoid transpositions in cat. * Bugfix. * Bugfix for the cuda kernel. * Add a benchmark. * Add more testing. * Test fix. * Faster kernel. * Add the missing kernel. * Tweak the test. * Add a metal kernel. * Fix for the metal kernel. * Get the tests to pass on metal. * Also use this opportunity to fix the metal kernel for ELU. * Add some bf16 kernels. * Clippy fixes.		2024-03-17 10:49:13 +01:00
..
basic_optimizer.rs	Add the optimizer trait. (#702)	2023-09-01 12:55:39 +01:00
cpu_benchmarks.rs	Optimize the cat operation on contiguous tensors (#1855)	2024-03-17 10:49:13 +01:00