diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-07 19:31:45 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-07 18:31:45 +0100 |
| commit | fc265d9dcfc13ee0b03f4a09537a9e7156b29231 (patch) | |
| tree | c94485d92647d340cb6e2a1eebc2c3aad4692d60 /candle-core/src/cuda_backend.rs | |
| parent | 2345b8ce3f8ebab6e04d6ea25f7c809efb037995 (diff) | |
| download | candle-fc265d9dcfc13ee0b03f4a09537a9e7156b29231.tar.gz candle-fc265d9dcfc13ee0b03f4a09537a9e7156b29231.tar.bz2 candle-fc265d9dcfc13ee0b03f4a09537a9e7156b29231.zip | |
Some CLIP fixes for stable diffusion. (#338)
* Some CLIP fixes for stable diffusion.
* Add the avg-pool2d operation on cpu.
Diffstat (limited to 'candle-core/src/cuda_backend.rs')
| -rw-r--r-- | candle-core/src/cuda_backend.rs | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/candle-core/src/cuda_backend.rs b/candle-core/src/cuda_backend.rs index 7b4b358d..e71ecfce 100644 --- a/candle-core/src/cuda_backend.rs +++ b/candle-core/src/cuda_backend.rs @@ -1381,6 +1381,10 @@ impl BackendStorage for CudaStorage { Ok(Self { slice, device }) } + fn avg_pool2d(&self, _: &Layout, _: (usize, usize), _: (usize, usize)) -> Result<Self> { + todo!() + } + fn index_select(&self, ids: &Self, l: &Layout, ids_l: &Layout, dim: usize) -> Result<Self> { let device = self.device().clone(); let slice = IndexSelect(ids, ids_l, dim).map(&self.slice, &device, l)?; |