fixes

Author: barakugav

jix/src/array.rs

@@ -790,7 +790,7 @@ impl<S: ArrayStorage> Array<S> {
             } else {
                 // Custom layout hint
                 let alloc_shape =
-                    S::Dimension::from_fn(ndim, |i| out_shape[layout_order[i] as usize] as u64);
+                    S::Dimension::from_fn(ndim, |i| out_shape[layout_order[i] as usize]);
                 let mut inv_order = S::Dimension::from_fn(ndim, |_| 0);
                 for dim in 0..ndim {
                     inv_order[layout_order[dim] as usize] = dim as u64;

jix/src/storage/params.rs

@@ -804,7 +804,7 @@ fn max_cost_orders(inputs: &[(f32, &[DimIdx], &[DimIdx])]) -> (DimArray<DimIdx>,
         .iter()
         .map(|(cost, _, _)| cost)
         .enumerate()
-        .max_by(|(i_a, a), (i_b, b)| a.partial_cmp(&b).unwrap().then(i_a.cmp(i_b)))
+        .max_by(|(i_a, a), (i_b, b)| a.partial_cmp(b).unwrap().then(i_a.cmp(i_b)))
         .unwrap()
         .0;
     let (_, scale_order, layout_order) = inputs[max_cost_index];