Laurent Mazare 1df2bddccf Add the layernorm specialized op. (#2212) ...

* Add the layernorm cuda kernels.

* Dedicated layer norm op.

* Add the slower variant.

* Plug the cuda implementation.

* Add the metal variant.

* Add a dedicated test.

* Bugfix.

2024-05-24 15:58:01 +02:00

..

affine.cu

Cuda backend optimization (#1886)

2024-03-20 18:32:55 +01:00

binary_op_macros.cuh

Cuda backend optimization (#1886)

2024-03-20 18:32:55 +01:00

binary.cu

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

cast.cu

Add cast_bf16_x/cast_x_bf16 when CUDA_ARCH<800 but CUDA_VERSION >= 11000 (#1919)

2024-03-23 13:44:10 +01:00

compatibility.cuh

Compat windows.

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

conv.cu

More efficient cuda implementation for ConvTranspose1d. (#2211)

2024-05-24 11:05:43 +02:00

cuda_utils.cuh

Relax the contiguous check for cuda kernels. (#2000)

2024-04-03 09:02:38 +02:00

fill.cu

Optimize the cat operation on contiguous tensors (#1855)

2024-03-17 10:49:13 +01:00

indexing.cu

Support scatter/index_add with i64 indices for f16 (#1915)

2024-03-22 11:51:41 +01:00

lib.rs

Add argsort. (#2132)

2024-04-27 20:17:35 +02:00

quantized.cu

Add the cuda dequantize f16 kernels. (#2137)

2024-04-28 20:05:05 +02:00

reduce.cu

Add the layernorm specialized op. (#2212)

2024-05-24 15:58:01 +02:00

sort.cu

Add argsort. (#2132)

2024-04-27 20:17:35 +02:00

ternary.cu

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

unary.cu

Fix sigmoid gradient calculation and move sigmoid into a specialized op (#2114)

2024-04-29 11:04:43 +02:00