5 Commits

Author	SHA1	Message	Date
Laurent Mazare	b219903d0f	Cuda backend optimization (#1886) ... * Attempt at making the kernel faster. * Also adapt the cast kernels. * Also apply to binary ops.	2024-03-20 18:32:55 +01:00
Laurent Mazare	9a5c7db91a	Add support for i64 (#563) ... * Add the i64 dtype. * Adapt the cuda kernels.	2023-08-23 10:42:19 +01:00
laurent	8ad47907f3	Add the kernels.	2023-06-30 10:26:56 +01:00
laurent	ec79fc43f2	Add the bf16 cuda kernels.	2023-06-29 23:12:02 +01:00
Nicolas Patry	d7f729fb8f	Refactor the hierarchy.	2023-06-27 11:57:27 +02:00