summaryrefslogtreecommitdiff
path: root/candle-nn/src/rotary_emb.rs
Commit message (Collapse)AuthorAgeFilesLines
* Update docs (#2553)zachcp2024-11-111-0/+2
| | | | | * add module docs for candle-core * doc each of the candle-nn modules and add the links to the doc page
* Add the rope THD kernel. (#2014)Laurent Mazare2024-04-051-0/+231
| | | | | | | | | * Add the rope THD kernel. * Cuda kernel for rope-thd. * Add the metal kernels. * Add a dedicated test.
* Fix detail in new RoPE implementation (#1935)Hugo Abonizio2024-03-251-1/+1
|
* Contiguous variant of the rope kernel. (#1929)Laurent Mazare2024-03-251-0/+252
| | | | | | | * Contiguous variant of the rope kernel. * Add the cuda kernel. * Metal kernel.
* Fast kernels for rotary embeddings. (#1928)Laurent Mazare2024-03-241-0/+247
* Fast kernels for rotary embeddings. * Add a test for the fast CPU kernel. * Rope cuda bindings. * Cuda kernel. * Metal kernel (part 1). * Cuda kernels. * Finish the metal kernel. * Use the new kernels in the quantized example. * Fix warning.