huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expose the storage publicly. (#157)

Browse Source
This commit is contained in:
Laurent Mazare
2023-07-13 13:52:36 +01:00
committed by GitHub
parent 21aa29ddce
commit 7adc8c903a
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-core/src/tensor.rs
View File

@ -1531,6 +1531,12 @@ impl Tensor {
fn storage(&self) -> std::sync::RwLockReadGuard<'_, Storage> { fn storage(&self) -> std::sync::RwLockReadGuard<'_, Storage> {
self.storage.read().unwrap() self.storage.read().unwrap()
} }
/// The storage used by this tensor, together with the layout to use to access it safely.
pub fn storage_and_layout(&self) -> (std::sync::RwLockReadGuard<'_, Storage>, &Layout) {
let storage = self.storage.read().unwrap();
(storage, &self.layout)
}
} }
macro_rules! bin_trait { macro_rules! bin_trait {