56 Commits

Author	SHA1	Message	Date
laurent	8ad47907f3	Add the kernels.	2023-06-30 10:26:56 +01:00
laurent	6486a6d7b2	Avoid some cast kernels.	2023-06-29 23:23:44 +01:00
laurent	ec79fc43f2	Add the bf16 cuda kernels.	2023-06-29 23:12:02 +01:00
laurent	1ce3843cab	Add the relu op.	2023-06-28 09:38:54 +01:00
laurent	380d61e990	Fix two cuda bugs (matmul and where_cond).	2023-06-27 11:31:04 +01:00
Nicolas Patry	d7f729fb8f	Refactor the hierarchy.	2023-06-27 11:57:27 +02:00