fixes

mimiquate · Jan 16, 2024 · 204bf6a · 204bf6a
1 parent 2c10d36
commit 204bf6a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/native/candlex/src/ops.rs b/native/candlex/src/ops.rs
@@ -460,13 +460,16 @@ macro_rules! custom_binary_bool_op {
                 s2: &MetalStorage,
                 _l2: &Layout,
             ) -> Result<(MetalStorage, Shape), candle_core::Error> {
-                let device = storage1.device();
+                use crate::metal_kernels;
+                use candle_core::DType;
+
+                let device = s1.device();
                 let shape = l1.shape();
                 let elem_count = shape.elem_count();
                 let command_buffer = device.command_buffer()?;
                 let output_buffer = device.new_buffer(elem_count, DType::U8, $name)?;
 
-                candle_metal_kernels::call_custom_binary_contiguous(
+                metal_kernels::call_custom_binary_contiguous(
                     &device.device,
                     &command_buffer,
                     metal_kernels::custom_binary::contiguous::logical_or::FLOAT,