Laurent Mazare c84883ecf2 Add a cuda kernel for upsampling. (#441) ...

* Add a cuda kernel for upsampling.

* Update for the latest tokenizers version.

2023-08-14 13:12:17 +01:00

..

affine.cu

Add the kernels.

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

binary_op_macros.cuh

Add some cmp tests. (#233)

2023-07-24 16:53:45 +01:00

binary.cu

Add some cmp tests. (#233)

2023-07-24 16:53:45 +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

Compat windows.

2023-08-10 17:46:47 +02:00

conv.cu

Add a cuda kernel for upsampling. (#441)

2023-08-14 13:12:17 +01:00

cuda_utils.cuh

Cuda support for the mnist training. (#277)

2023-07-29 19:48:04 +01:00

fill.cu

Refactor the hierarchy.

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

indexing.cu

Remove the embedding ops in favor of index-select. (#299)

2023-08-02 05:42:11 +01:00

lib.rs

Cuda kernels for IndexAdd/ScatterAdd. (#236)

2023-07-24 21:53:08 +01:00

reduce.cu

Cuda support for the mnist training. (#277)

2023-07-29 19:48:04 +01:00

ternary.cu

Support for where-cond on cuda for u8 and u32. (#274)

2023-07-29 11:48:58 +01:00

unary.cu

Add the recip op + use it in stable-diffusion. (#331)

2023-08-06 21:14:52 +01:00