huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 12:20:46 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,147 Commits 74 Branches 16 Tags
18d6db2180800dcc134ffabe8523a774c6a7f9a3
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