Laurent Mazare a4c56a958e Add the const-set op. (#2910) ...

* Add the const-set op.

* Cuda implementation.

* Bugfix.

* Metal cleanup.

* Add the metal kernels.

* Add some testing.

* Finish the metal implementation.

* Bump the version.

2025-04-19 10:07:02 +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

Im2col cuda optimization. (#2885)

2025-04-13 10:07:53 +02:00

cuda_utils.cuh

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

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

fill.cu

Add the const-set op. (#2910)

2025-04-19 10:07:02 +02:00

indexing.cu

Check the bounds in the cuda indexing kernels. (#2908)

2025-04-18 20:08:17 +02:00

lib.rs

Update to cudarc 0.14 (breaking change). (#2858)

2025-04-03 09:12:19 +02:00

ptx.rs

Update to cudarc 0.14 (breaking change). (#2858)

2025-04-03 09:12:19 +02:00

quantized.cu

Import the ggml_cuda_dp4a function. (#2628)

2024-11-19 03:41:34 +01:00

reduce.cu

Improved launch config for layer-norm/rms-norm. (#2591)

2024-11-04 10:42:18 +01: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