Laurent Mazare bc3be6f9b0 Add the elu cuda kernel. (#114)

2023-07-10 07:57:01 +01:00

..

affine.cu

Add the kernels.

2023-06-30 10:26:56 +01:00

binary_op_macros.cuh

Refactor the hierarchy.

2023-06-27 11:57:27 +02:00

binary.cu

Add the kernels.

2023-06-30 10:26:56 +01:00

cast.cu

Bugfix: remove the u8/bf16 conversion kernel as it is ambiguous.

2023-06-30 10:43:32 +01:00

compatibility.cuh

Add the bf16 cuda kernels.

2023-06-29 23:12:02 +01:00

conv.cu

Make it easier to use whisper samples from the repo. (#112)

2023-07-08 18:48:27 +01:00

cuda_utils.cuh

Add the bf16 cuda kernels.

2023-06-29 23:12:02 +01:00

embeddings.cu

Add the kernels.

2023-06-30 10:26:56 +01:00

fill.cu

Refactor the hierarchy.

2023-06-27 11:57:27 +02:00

lib.rs

Cuda kernel for the conv1d op (#111)

2023-07-08 18:13:25 +01:00

reduce.cu

Sketch a fast cuda kernel for reduce-sum. (#109)

2023-07-08 12:43:56 +01:00

ternary.cu

Add the kernels.

2023-06-30 10:26:56 +01:00

unary.cu

Add the elu cuda kernel. (#114)

2023-07-10 07:57:01 +01:00