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
945 Commits 74 Branches 16 Tags
431051cc327c22733ecf20c80bcbf1760444a751
Commit Graph

1 Commits

Author SHA1 Message Date
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