Expose the synchronize function on the generic device.

huggingface · Apr 14, 2024 · 227a2b0 · 227a2b0
1 parent 4ecedb1
commit 227a2b0
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/candle-core/src/device.rs b/candle-core/src/device.rs
@@ -337,4 +337,12 @@ impl Device {
  }
  }
  }
+
+ pub fn synchronize(&self) -> Result<()> {
+ match self {
+ Self::Cpu => Ok(()),
+ Self::Cuda(d) => d.synchronize(),
+ Self::Metal(d) => d.synchronize(),
+ }
+ }
 }