diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-06-29 10:14:12 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-06-29 10:14:12 +0100 |
| commit | c8fc9da73701e3f5d16695d6a38d55d471c7b82c (patch) | |
| tree | d517caf2aedb654261a3dccadb95eeb4badc958e /candle-core/src/tensor.rs | |
| parent | eda46d2df20f1b6186473e8ee7a26c2d36aa05df (diff) | |
| parent | c9c468e1aaf0ce071b145f15aba830e9600fd6e6 (diff) | |
| download | candle-c8fc9da73701e3f5d16695d6a38d55d471c7b82c.tar.gz candle-c8fc9da73701e3f5d16695d6a38d55d471c7b82c.tar.bz2 candle-c8fc9da73701e3f5d16695d6a38d55d471c7b82c.zip | |
Merge pull request #33 from LaurentMazare/cuda-map
Simplify the dtype matchings in the cuda backend
Diffstat (limited to 'candle-core/src/tensor.rs')
| -rw-r--r-- | candle-core/src/tensor.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/src/tensor.rs b/candle-core/src/tensor.rs index f64bd6f2..4b9b3306 100644 --- a/candle-core/src/tensor.rs +++ b/candle-core/src/tensor.rs @@ -709,7 +709,7 @@ impl Tensor { pub fn copy(&self) -> Result<Tensor> { let tensor_ = Tensor_ { id: TensorId::new(), - storage: Arc::new(self.storage.try_clone()?), + storage: Arc::new(self.storage.try_clone(self.layout())?), layout: self.layout.clone(), op: None, // TODO is_variable: false, |