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-core/src/cpu_backend.rs | |
| 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-core/src/cpu_backend.rs')
| -rw-r--r-- | candle-core/src/cpu_backend.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/src/cpu_backend.rs b/candle-core/src/cpu_backend.rs index c39cb9f7..59c17387 100644 --- a/candle-core/src/cpu_backend.rs +++ b/candle-core/src/cpu_backend.rs @@ -244,7 +244,7 @@ impl ReduceIndex { val = s } } - dst[unstr_index] = g(val, acc) + dst_to_set[unstr_index] = g(val, acc) } } } |