Laurent Mazare 34f4b3187e Add a naive conv2d cuda kernel. (#438) ...

* Add a naive conv2d cuda kernel.

* Proper conv2d support on the rust side.

* Conv1d testing on gpu.

* Also use the test on gpus.

* Fix the clean-ptx target.

2023-08-14 10:34:42 +01:00

..

conv_tests.rs

Add a naive conv2d cuda kernel. (#438)

2023-08-14 10:34:42 +01:00

custom_op_tests.rs

Rename the candle crate to candle-core (#301)

2023-08-02 08:20:22 +01:00

display_tests.rs

Rename the candle crate to candle-core (#301)

2023-08-02 08:20:22 +01:00

grad_tests.rs

More accelerate optimizations (#427)

2023-08-13 12:53:34 +01:00

indexing_tests.rs

Rename the candle crate to candle-core (#301)

2023-08-02 08:20:22 +01:00

layout_tests.rs

Rename the candle crate to candle-core (#301)

2023-08-02 08:20:22 +01:00

pool_tests.rs

Upsample test + bugfix. (#399)

2023-08-10 21:02:35 +02:00

tensor_tests.rs

Further randn tweaks: use the appropriate rng rather than the f64 one, some cleanup. (#383)

2023-08-10 05:48:19 +01:00

test_utils.rs

More accelerate optimizations (#427)

2023-08-13 12:53:34 +01:00