Laurent Mazare 90374097dc Cudnn support (#445) ...

* Add a cudnn feature to be used for conv2d.

* Allocate the proper workspace.

* Only create a single cudnn handle per cuda device.

* Proper cudnn usage.

* Bugfix.

2023-08-14 21:30:41 +01:00

..

basics.rs

Add some conv1d test + bugfix using padding. (#349)

2023-08-08 20:50:20 +01:00

cpu_benchmarks.rs

Add a softmax bench. (#433)

2023-08-13 20:09:18 +01:00

cuda_basics.rs

Cudnn support (#445)

2023-08-14 21:30:41 +01:00

cuda_sum_benchmark.rs

More accelerate optimizations (#427)

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