huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-22 04:22:50 +00:00
Code Issues Packages Projects Releases Wiki Activity
659 Commits 74 Branches 16 Tags
d2dea11ef65566518b6d722e9dbe33822aa98b74
Commit Graph

2 Commits

Author SHA1 Message Date
Laurent Mazare
944d70bd9a Add a test for scatter add. (#238) 
* Add a test for scatter add (segfaults on gpus for now).

* Bugfix for the scatter add cuda kernel.
 2023-07-25 09:12:14 +01:00
Laurent Mazare
74a6a769dd Cuda kernels for IndexAdd/ScatterAdd. (#236) 
* Skeleton methods for IndexAdd/ScatterAdd.

* Add a Map2InPlace trait.

* Add the glue code for the index-add/scatter-add kernels.

* Tweak the file name: embeddings -> indexing.

* Add the cuda kernel for indexadd.

* And add the scatter-add kernels.
 2023-07-24 21:53:08 +01:00