diff options
Diffstat (limited to 'candle-core/src/metal_backend.rs')
-rw-r--r-- | candle-core/src/metal_backend.rs | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/candle-core/src/metal_backend.rs b/candle-core/src/metal_backend.rs index d38796a1..b8b951f0 100644 --- a/candle-core/src/metal_backend.rs +++ b/candle-core/src/metal_backend.rs @@ -113,21 +113,23 @@ impl MetalDevice { self._new_buffer(size, MTLResourceOptions::StorageModePrivate, name) } - fn _new_buffer(&self, size: NSUInteger, option: MTLResourceOptions, name: &str) -> Arc<Buffer> { - // println!("Creating new buffer {name}"); + fn _new_buffer( + &self, + size: NSUInteger, + option: MTLResourceOptions, + _name: &str, + ) -> Arc<Buffer> { let mut buffers = self.buffers.try_write().unwrap(); let subbuffers = buffers.entry((size, option)).or_insert(vec![]); for sub in &mut *subbuffers { if Arc::strong_count(sub) == 1 { - // println!("Reusing tensor {size} {name}"); return sub.clone(); } } let new_buffer = self.device.new_buffer(size as NSUInteger, option); let new_buffer = Arc::new(new_buffer); subbuffers.push(new_buffer.clone()); - // println!("Created tensor {size} {name}"); for subbuffers in buffers.values_mut() { let newbuffers = subbuffers .iter() |