diff --git a/candle-core/src/cpu_backend.rs b/candle-core/src/cpu_backend.rs index ed3dd3fc..c3ec8249 100644 --- a/candle-core/src/cpu_backend.rs +++ b/candle-core/src/cpu_backend.rs @@ -445,7 +445,7 @@ pub fn unary_map_vec U, FV: FnMut(&[T], &mut [U } // This function maps over two strided index sequences. -fn binary_map U>( +pub fn binary_map U>( lhs_l: &Layout, rhs_l: &Layout, lhs: &[T], @@ -525,7 +525,7 @@ fn binary_map U>( } // Similar to binary_map but with vectorized variants. -fn binary_map_vec T, FV: FnMut(&[T], &[T], &mut [T])>( +pub fn binary_map_vec T, FV: FnMut(&[T], &[T], &mut [T])>( lhs_l: &Layout, rhs_l: &Layout, lhs: &[T],