We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5514fd5 commit 98c0436Copy full SHA for 98c0436
1 file changed
candle-core/src/metal_backend/device.rs
@@ -223,6 +223,16 @@ impl MetalDevice {
223
self.allocate_buffer(size, METAL_SHARED_BUFFER_STORAGE_MODE, name)
224
}
225
226
+ pub fn new_buffer_private(
227
+ &self,
228
+ element_count: usize,
229
+ dtype: DType,
230
+ name: &str,
231
+ ) -> Result<Arc<Buffer>> {
232
+ let size = (element_count * dtype.size_in_bytes()) as NSUInteger;
233
+ self.allocate_buffer(size, metal::MTLResourceOptions::StorageModePrivate, name)
234
+ }
235
+
236
/// Creates a new buffer (not necessarily zeroed).
237
/// The buffer is [MTLManaged](https://developer.apple.com/documentation/metal/mtlstoragemode)
238
/// This means the buffer can be read on the CPU but will require manual
0 commit comments