Laurent Mazare 944d70bd9a Add a test for scatter add. (#238) ...

* Add a test for scatter add (segfaults on gpus for now).

* Bugfix for the scatter add cuda kernel.

2023-07-25 09:12:14 +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

Add a test for scatter add. (#238)

2023-07-25 09:12:14 +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