laurent 97d9142dee Add a first kernel.

2023-06-21 20:48:22 +01:00

..

cpu_backend.rs

Move more bits to the backend part.

2023-06-21 10:34:51 +01:00

cuda_backend.rs

Add a first kernel.

2023-06-21 20:48:22 +01:00

device.rs

Move the data between the host and the device.

2023-06-21 19:43:25 +01:00

dtype.rs

Avoid some duplication using a macro + add some basic example to make debugging easier.

2023-06-21 10:08:41 +01:00

error.rs

Use a type alias for cuda errors.

2023-06-21 19:50:00 +01:00

lib.rs

Move the data between the host and the device.

2023-06-21 19:43:25 +01:00

op.rs

Add a couple operators.

2023-06-20 22:32:11 +01:00

shape.rs

Add some very basic backprop.

2023-06-20 20:33:44 +01:00

storage.rs

Add a first kernel.

2023-06-21 20:48:22 +01:00

strided_index.rs

Move the StridedIndex in its own module.

2023-06-21 07:44:36 +01:00

tensor.rs

Use a reference for the device.

2023-06-21 19:55:57 +01:00