Thomas Santerre c5626b8271 Add support for "sign" on tensors (#2012) ...

* add the sign unary operator

* remove uneeded import

* remove uneeded import

* undo formatting

* undo formatting

* remove unnecessary redefintion

* allow gradient to flow through for sign and round

* fix cpu ops to ensure that negzero and positive zero are handled properly

* clippy fixes

* Properly avoid gradient tracking.

* Use a branchless version.

---------

Co-authored-by: laurent <laurent.mazare@gmail.com>

2024-04-04 22:32:47 +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

ConvTranspose1d cuda support. (#1697)

2024-02-12 15:03:18 +01: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

Cuda acceleration for quantized model. (#1754)

2024-02-25 18:11:47 +01:00

quantized.cu

More ggml cuda kernels (#1977)

2024-04-01 00:15:48 +02:00

reduce.cu

Use the new rope kernel in mistral. (#1937)

2024-03-25 23:26:05 +01:00

ternary.cu

Add support for i64 (#563)

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

unary.cu

Add support for "sign" on tensors (#2012)

2024-04-04 22:32:47 +02:00