summaryrefslogtreecommitdiff
path: root/candle-nn/examples/cpu_benchmarks.rs
Commit message (Collapse)AuthorAgeFilesLines
* Implement the module trait directly for QMatMul. (#1372)Laurent Mazare2023-11-251-1/+1
|
* Add a matvec cpu benchmark. (#1076)Laurent Mazare2023-10-121-3/+22
|
* Convmixer (#1073)Laurent Mazare2023-10-111-2/+2
| | | | | | | | | | | * Only optimize float tensors. * Use full tensors for zeros and ones. * Add a benchmark for the matmul slowness. * Add the convmixer model. * Proper adaptive pooling.
* Improve the quantized whisper setup. (#1018)Laurent Mazare2023-10-021-1/+1
| | | | | | | * Improve the quantized whisper setup. * Fix the config file paths. * Use the standard matmul where possible.
* Bugfix for the conv2d cpu kernel. (#820)Laurent Mazare2023-09-111-1/+1
|
* im2col based conv2d (#802)Laurent Mazare2023-09-101-16/+69
| | | | | | | | | | | | | * im2col implementation for conv2d. * Fix for the im2col implementation to match the current conv2d. * Small optimization. * Add a cuda kernel. * Handle arbitrary layouts. * Im2Col cuda code.
* Bugfix so that im2col produce the same results as conv2d. (#801)Laurent Mazare2023-09-101-1/+5
|
* Add an im2col based benchmark. (#800)Laurent Mazare2023-09-101-2/+71
| | | | | * Add an im2col based benchmark. * Reshape the final result.
* Add a custom softmax implementation. (#744)Laurent Mazare2023-09-051-0/+176
* 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.