Fix build bugs with metal, NSUInteger (#27)

EricLBuehler · Sep 25, 2024 · 105dbd0 · 105dbd0
1 parent 9e31a19
commit 105dbd0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/candle-core/src/quantized/metal.rs b/candle-core/src/quantized/metal.rs
@@ -208,7 +208,9 @@ impl QMetalStorage {
     }
 
     pub fn data(&self) -> Result<Vec<u8>> {
-        let size = (self.count * self.dtype.size_in_bytes()) as NSUInteger;
+        use metal::NSUInteger;
+        let count = self.buffer.length() as usize;
+        let size = (count * self.dtype.type_size()) as NSUInteger;
 
         let buffer = self.device.new_buffer_managed(size)?;
         {
@@ -220,7 +222,7 @@ impl QMetalStorage {
             blit.end_encoding();
         }
         self.device.wait_until_completed()?;
-        Ok(read_to_vec::<u8>(&buffer, self.count))
+        Ok(read_to_vec::<u8>(&buffer, count))
     }
 }