laurent f2ac5547fc Avoid the race condition on cuda sums.

2023-06-26 16:19:06 +01:00

..

affine.cu

Optimize the unary cuda kernels for the contiguous case.

2023-06-23 18:40:15 +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

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

Add a cuda kernel for embeddings.

2023-06-26 11:47:57 +01:00

fill.cu

Integrate the kernels bits.

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

lib.rs

Add the where kernels.

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

reduce.cu

Avoid the race condition on cuda sums.

2023-06-26 16:19:06 +01:00

ternary.cu

Add the where kernels.

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

unary.cu

Add a couple unitary ops.

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