huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 19:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,142 Commits 74 Branches 16 Tags
b2796ce6efb62f58c2dc9725585e48c5959e8221
Commit Graph

5 Commits

Author SHA1 Message Date
Laurent Mazare
9a5c7db91a Add support for i64 (#563) 
* Add the i64 dtype.

* Adapt the cuda kernels.
 2023-08-23 10:42:19 +01:00
Laurent Mazare
c950a5c6b1 Cuda support for the mnist training. (#277) 
* Cuda support for the mnist training.

* min/max fix + testing.

* Add the argmin/argmax tests.

* More cuda support for argmin/argmax.

* Cuda kernels for argmin and argmax.
 2023-07-29 19:48:04 +01:00
Laurent Mazare
536c5e702e Cuda kernels for fast min/max reductions (#203) 
* Add the min/max cuda kernels.

* Better integration of the cuda kernels.
 2023-07-19 18:12:27 +01:00
laurent
ec79fc43f2 Add the bf16 cuda kernels. 2023-06-29 23:12:02 +01:00
Nicolas Patry
d7f729fb8f Refactor the hierarchy. 2023-06-27 11:57:27 +02:00