Laurent Mazare 581b104f97 Indexing cuda (#235) ...

* Allow using uint8_t for indexing.

* Revert the default cuda feature.

* Add a cuda-kernel for index-select.

* Add a test for gather.

2023-07-24 20:22:47 +01:00

..

custom_op_tests.rs

Add binary and ternary custom ops. (#217)

2023-07-21 17:29:50 +01:00

display_tests.rs

Add some display tests + bugfixes.

2023-06-27 21:37:28 +01:00

grad_tests.rs

Add some epsilon tolerance to grad tests so that they work on cuda / mkl. (#213)

2023-07-21 12:45:14 +01:00

indexing_tests.rs

Test the index op + bugfix. (#148)

2023-07-12 15:42:36 +01:00

layout_tests.rs

Iteration over strided blocks (#175)

2023-07-15 21:30:35 +01:00

tensor_tests.rs

Indexing cuda (#235)

2023-07-24 20:22:47 +01:00

test_utils.rs

Add some epsilon tolerance to grad tests so that they work on cuda / mkl. (#213)

2023-07-21 12:45:14 +01:00