huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,146 Commits 74 Branches 16 Tags
4f18180fc7564b9a02cab13c4acda0ac7b17a799
Commit Graph

3 Commits

Author SHA1 Message Date
Laurent Mazare
4f18180fc7 Bugfix so that im2col produce the same results as conv2d. (#801) 2023-09-10 16:59:46 +01:00
Laurent Mazare
559944146f Add an im2col based benchmark. (#800) 
* Add an im2col based benchmark.

* Reshape the final result.
 2023-09-10 16:56:28 +01:00
Laurent Mazare
1c9e5394a5 Add a custom softmax implementation. (#744) 
* Add a custom softmax implementation.

* Add softmaxlastdim to the benchmarks.

* And add a test.

* Support more dtypes.

* Polish the code.

* Use the slow implementation on cuda.

* Add a todo for the cuda kernel.
 2023-09-05 14:20:23 +01:00