summaryrefslogtreecommitdiff
path: root/src/tensor.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/tensor.rs')
-rw-r--r--src/tensor.rs3
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!(),
}
}