|
|
ee3d290f8b
|
Cuda support for dtype conversions.
|
2023-06-27 09:15:46 +01:00 |
|
|
|
cd2a171c06
|
Add the where kernels.
|
2023-06-26 13:25:02 +01:00 |
|
|
|
f6104c4b64
|
Add the reduce-sum kernel.
|
2023-06-26 12:35:26 +01:00 |
|
|
|
16f0f5b9d2
|
Add a cuda kernel for embeddings.
|
2023-06-26 11:47:57 +01:00 |
|
|
|
5276755fb3
|
Add cuda support for unary ops.
|
2023-06-22 15:12:59 +01:00 |
|
|
|
b8f514d9c6
|
Add more binary kernels.
|
2023-06-22 14:07:02 +01:00 |
|
|
|
4b1c3405e9
|
Add a couple cuda kernels from dfdx.
|
2023-06-22 12:56:29 +01:00 |
|
|
|
083ced4428
|
Integrate the kernels bits.
|
2023-06-22 09:59:00 +01:00 |
|
