Laurent Mazare d0aa197b07 ConvTranspose1d cuda support. (#1697) ...

* ConvTranspose1d cuda support.

* Add the conv-transpose1d kernel.

* Remove some unused variables.

2024-02-12 15:03:18 +01:00

..

affine.cu

Add support for i64 (#563)

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

binary_op_macros.cuh

Add some cmp tests. (#233)

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

binary.cu

Add support for i64 (#563)

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

cast.cu

Rework the cuda casting bits. (#1112)

2023-10-17 09:44:51 +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

Add the rounding operators. (#1030)

2023-10-04 17:58:44 +01:00

fill.cu

fix: add missing gpu fill_* (#996)

2023-09-29 15:49:30 +01:00

indexing.cu

fix: fix index_select cuda kernel for src target dim different than ids dim when selecting dim > 0 (#1037)

2023-10-05 18:46:13 +01:00

lib.rs

Cuda kernels for IndexAdd/ScatterAdd. (#236)

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

reduce.cu

Add a dedicated cuda kernel for softmax. (#746)

2023-09-05 17:53:20 +02:00

ternary.cu

Add support for i64 (#563)

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

unary.cu

Add the rounding operators. (#1030)

2023-10-04 17:58:44 +01:00