From 5e241722cb246b9b1011c1d9ad8688b167e68726 Mon Sep 17 00:00:00 2001
From: Mikko Juola <mikjuo@gmail.com>
Date: Mon, 20 Mar 2023 18:30:41 -0700
Subject: [PATCH] Fix compilation when opencl feature is being used.

---
 src/tensor.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/tensor.rs b/src/tensor.rs
index d36c476..f026161 100644
--- a/src/tensor.rs
+++ b/src/tensor.rs
@@ -2019,7 +2019,7 @@ impl Tensor {
         if data.is_null() {
             panic!("to_cpu_inplace: Failed to allocate tensor");
         }
-        TENSORS_BYTES_ALLOCATED.fetch_add(layout.size(), std::sync::atomic::Ordering::Relaxed);
+        TENSORS_BYTES_ALLOCATED.fetch_add(self.layout.size(), std::sync::atomic::Ordering::Relaxed);
         let ev = od.as_mut().unwrap().data_u16_from_gpu(data as *mut u16)?;
         self.data = data as *mut u16 as *mut u8;
         self.waiting_for_data = Some(ev);