huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 20:22:49 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
90374097dc99b14dfc935318a18c21fc5909291f
candle/candle-core
History
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
..
examples
Cudnn support (#445)
2023-08-14 21:30:41 +01:00
src
Cudnn support (#445)
2023-08-14 21:30:41 +01:00
tests
Add a cuda kernel for upsampling. (#441)
2023-08-14 13:12:17 +01:00
Cargo.toml
Cudnn support (#445)
2023-08-14 21:30:41 +01:00
LICENSE
Refactor the hierarchy.
2023-06-27 11:57:27 +02:00
README.md
Refactor the hierarchy.
2023-06-27 11:57:27 +02:00

README.md

candle

Minimalist ML framework for Rust