diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-05-24 15:58:01 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-24 15:58:01 +0200 |
| commit | 1df2bddccfbb4ab511a8cc3a87476d1fa72416bc (patch) | |
| tree | 3633bc51e3bac3d542d9dfe06d509db20f5374e9 /candle-nn/tests/layer_norm.rs | |
| parent | 6f0b807ffd553fed27325a2a118b0e30bb6d9cbd (diff) | |
| download | candle-1df2bddccfbb4ab511a8cc3a87476d1fa72416bc.tar.gz candle-1df2bddccfbb4ab511a8cc3a87476d1fa72416bc.tar.bz2 candle-1df2bddccfbb4ab511a8cc3a87476d1fa72416bc.zip | |
Add the layernorm specialized op. (#2212)
* Add the layernorm cuda kernels.
* Dedicated layer norm op.
* Add the slower variant.
* Plug the cuda implementation.
* Add the metal variant.
* Add a dedicated test.
* Bugfix.
Diffstat (limited to 'candle-nn/tests/layer_norm.rs')
0 files changed, 0 insertions, 0 deletions