binary strided i64

mimiquate · Jan 18, 2024 · 95efec7 · 95efec7
1 parent c8405b1
commit 95efec7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/native/candlex/src/metal_kernels.rs b/native/candlex/src/metal_kernels.rs
@@ -57,6 +57,7 @@ macro_rules! ops {
                     use super::Kernel;
 
                     pub const FLOAT: Kernel = Kernel(concat!(stringify!($name), "_f32_strided"));
+                    pub const I64: Kernel = Kernel(concat!(stringify!($name), "_i64_strided"));
                 }
             )+
         }

diff --git a/native/candlex/src/ops.rs b/native/candlex/src/ops.rs
@@ -413,6 +413,7 @@ macro_rules! custom_binary_op {
                 } else {
                     let kernel_name = match dtype {
                         DType::F32 => metal_kernels::custom_binary::strided::$name::FLOAT,
+                        DType::I64 => metal_kernels::custom_binary::strided::$name::I64,
                         dtype => {
                             candle_core::bail!("Metal strided custom binary {} {dtype:?} not implemented", stringify!($name))
                         }
-Original file line number
+Diff line change
@@ Expand Up / @@ -57,6 +57,7 @@ macro_rules! ops { @@
                         use super::Kernel;
                         pub const FLOAT: Kernel = Kernel(concat!(stringify!($name), "_f32_strided"));
+                        pub const I64: Kernel = Kernel(concat!(stringify!($name), "_i64_strided"));
                     }
                 )+
             }
@@ Expand Down @@