Skip to content

Commit 98c0436

Browse files
committed
Add new_buffer_private for metal
1 parent 5514fd5 commit 98c0436

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

candle-core/src/metal_backend/device.rs

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,16 @@ impl MetalDevice {
223223
self.allocate_buffer(size, METAL_SHARED_BUFFER_STORAGE_MODE, name)
224224
}
225225

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+
226236
/// Creates a new buffer (not necessarily zeroed).
227237
/// The buffer is [MTLManaged](https://developer.apple.com/documentation/metal/mtlstoragemode)
228238
/// This means the buffer can be read on the CPU but will require manual

0 commit comments

Comments
 (0)