Async tensor copying. (#1900)

This commit is contained in:
Laurent Mazare
2024-03-21 13:09:42 +01:00
committed by GitHub
parent bb3ee48039
commit ec97c98e81
7 changed files with 59 additions and 4 deletions

View File

@ -1867,6 +1867,10 @@ impl BackendDevice for MetalDevice {
))
}
fn storage_from_cpu_storage_owned(&self, storage: CpuStorage) -> Result<Self::Storage> {
self.storage_from_cpu_storage(&storage)
}
fn rand_uniform(
&self,
shape: &Shape,