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

Prepare for the custom-op extension. (#1892)

Browse Source
This commit is contained in:
Laurent Mazare
2024-03-21 07:02:20 +01:00
committed by GitHub
parent af7f8b87d3
commit 74b7f59261
5 changed files with 256 additions and 247 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
candle-core/src/storage.rs
View File

@ -1,6 +1,7 @@
use crate::backend::BackendStorage;
use crate::op::{self, CmpOp, CustomOp1, CustomOp2, CustomOp3, ReduceOp};
use crate::op::{self, CmpOp, ReduceOp};
use crate::{CpuStorage, CudaStorage, DType, Device, Error, Layout, MetalStorage, Result, Shape};
use crate::{CustomOp1, CustomOp2, CustomOp3};
// We do not want to implement Clone on Storage as cloning may fail because of
// out of memory. Instead try_clone should be used.