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

Add a synchronize method to devices. (#2055)

Browse Source 
* Add a synchronize method to devices.

* Metal version.
This commit is contained in:
Laurent Mazare
2024-04-14 16:32:55 +02:00
committed by GitHub
parent 50e49ecc5f
commit 53e5380bf6
6 changed files with 24 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-core/src/metal_backend/mod.rs
View File

@ -1790,6 +1790,10 @@ impl BackendDevice for MetalDevice {
Ok(())
}
fn synchronize(&self) -> Result<()> {
self.wait_until_completed()
}
}
fn read_to_vec<T: Clone>(buffer: &Buffer, n: usize) -> Vec<T> {