laurent bb262ecc99 More casting kernels.

2023-06-27 09:36:35 +01:00

..

affine.cu

Add the f16 affine kernel.

2023-06-26 22:05:31 +01:00

binary_op_macros.cuh

Also optimize the contiguous case for the binary cuda kernels.

2023-06-23 19:04:13 +01:00

binary.cu

More u32 support.

2023-06-23 14:54:03 +01:00

cast.cu

More casting kernels.

2023-06-27 09:36:35 +01:00

compatibility.cuh

Add a couple cuda kernels from dfdx.

2023-06-22 12:56:29 +01:00

cuda_utils.cuh

Add the where kernels.

2023-06-26 13:25:02 +01:00

embeddings.cu

Cuda support for embedding f16.

2023-06-26 21:58:15 +01:00

fill.cu

Integrate the kernels bits.

2023-06-22 09:59:00 +01:00

lib.rs

Cuda support for dtype conversions.

2023-06-27 09:15:46 +01:00

reduce.cu

Add the f16 sum kernel (fix).

2023-06-26 22:02:22 +01:00

ternary.cu

where_cond for f16.

2023-06-26 22:14:32 +01:00

unary.cu

Add a couple unitary ops.

2023-06-23 20:19:20 +01:00