summaryrefslogtreecommitdiff
path: root/candle-core/src/metal_backend.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-core/src/metal_backend.rs')
-rw-r--r--candle-core/src/metal_backend.rs10
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()