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

8 Commits

Author SHA1 Message Date
Laurent Mazare
94c6a8d3d3 Add a dedicated cuda kernel for softmax. (#746) 2023-09-05 17:53:20 +02:00
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
23827c49cd Cleanup some todos. (#226) 
* Cleanup some todos.

* Fix more todo.

* Optimize for the contiguous case.

* Add the IntDType trait.

* Handle the intdtype trait for more ops.

* Remove a todo.

* Remove a todo.
 2023-07-23 16:00:00 +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 Mazare
e676f85f00 Sketch a fast cuda kernel for reduce-sum. (#109) 
* Sketch a fast cuda kernel for reduce-sum.

* Sketch the rust support code for the fast sum kernel.

* More work on the fast kernel.

* Add some testing ground.

* A couple fixes for the fast sum kernel.
 2023-07-08 12:43:56 +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