huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 19:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

CPU implementation for upsample-nearest2d. (#339)

Browse Source
This commit is contained in:
Laurent Mazare
2023-08-07 21:07:10 +02:00
committed by GitHub
parent fc265d9dcf
commit d0d7010682
5 changed files with 61 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-core/src/dummy_cuda_backend.rs
View File

@ -123,6 +123,10 @@ impl crate::backend::BackendStorage for CudaStorage {
fn avg_pool2d(&self, _: &Layout, _: (usize, usize), _: (usize, usize)) -> Result<Self> {
Err(Error::NotCompiledWithCudaSupport)
}
fn upsample_nearest2d(&self, _: &Layout, _: usize, _: usize) -> Result<Self> {
Err(Error::NotCompiledWithCudaSupport)
}
}
impl crate::backend::BackendDevice for CudaDevice {