diff options
author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-07-29 19:48:04 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-07-29 19:48:04 +0100 |
commit | c950a5c6b1bd207f6f9cba18780e0baa44e3254a (patch) | |
tree | dda49e92ece97b85f9260eba714963d94f694525 /candle-wasm-examples/llama2-c | |
parent | 16c33383eb2beda515962b219728209b9edb2946 (diff) | |
download | candle-c950a5c6b1bd207f6f9cba18780e0baa44e3254a.tar.gz candle-c950a5c6b1bd207f6f9cba18780e0baa44e3254a.tar.bz2 candle-c950a5c6b1bd207f6f9cba18780e0baa44e3254a.zip |
Cuda support for the mnist training. (#277)
* Cuda support for the mnist training.
* min/max fix + testing.
* Add the argmin/argmax tests.
* More cuda support for argmin/argmax.
* Cuda kernels for argmin and argmax.
Diffstat (limited to 'candle-wasm-examples/llama2-c')
0 files changed, 0 insertions, 0 deletions