diff options
Diffstat (limited to 'src/tensor.rs')
-rw-r--r-- | src/tensor.rs | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/tensor.rs b/src/tensor.rs index b8fa738a..2d704a65 100644 --- a/src/tensor.rs +++ b/src/tensor.rs @@ -209,6 +209,7 @@ impl Tensor { let data = S::cpu_storage_as_slice(cpu_storage)?; Ok(data[0]) } + Storage::Cuda { .. } => todo!(), } } @@ -249,6 +250,7 @@ impl Tensor { let data = S::cpu_storage_as_slice(cpu_storage)?; Ok(self.strided_index().map(|i| data[i]).collect()) } + Storage::Cuda { .. } => todo!(), } } @@ -266,6 +268,7 @@ impl Tensor { assert!(src_index.next().is_none()); Ok(rows) } + Storage::Cuda { .. } => todo!(), } } |