diff options
| author | laurent <laurent.mazare@gmail.com> | 2023-07-21 20:12:48 +0100 |
|---|---|---|
| committer | laurent <laurent.mazare@gmail.com> | 2023-07-21 20:12:48 +0100 |
| commit | 27174a82aaef01f2457e80b0d5ed2a8d8c04dd28 (patch) | |
| tree | edbfa21e24a417f9533ef3cc375854eb0b322ab7 /candle-core/src/cuda_backend.rs | |
| parent | 5cc843550d454f4e8143a915add7cb16db7d05cc (diff) | |
| download | candle-27174a82aaef01f2457e80b0d5ed2a8d8c04dd28.tar.gz candle-27174a82aaef01f2457e80b0d5ed2a8d8c04dd28.tar.bz2 candle-27174a82aaef01f2457e80b0d5ed2a8d8c04dd28.zip | |
Start adding index-add.
Diffstat (limited to 'candle-core/src/cuda_backend.rs')
| -rw-r--r-- | candle-core/src/cuda_backend.rs | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/candle-core/src/cuda_backend.rs b/candle-core/src/cuda_backend.rs index cdbfd0c6..a5633836 100644 --- a/candle-core/src/cuda_backend.rs +++ b/candle-core/src/cuda_backend.rs @@ -1064,6 +1064,17 @@ impl BackendStorage for CudaStorage { fn index_select(&self, _: &Self, _: &Layout, _: &Layout, _: usize) -> Result<Self> { Err(CudaError::InternalError("TODO: implement index-select").into()) } + fn index_add( + &self, + _: &Layout, + _: &Self, + _: &Layout, + _: &Self, + _: &Layout, + _: usize, + ) -> Result<Self> { + Err(CudaError::InternalError("TODO: implement index-add").into()) + } fn matmul( &self, |