fmt/fix

Author: crutcher

crates/burn-fusion/src/ops/float.rs

@@ -8,11 +8,11 @@ use crate::{
     unary_float_ops,
 };
 use burn_ir::{
-    BackendIr, BaseOperationIr, BinaryOpIr, CatOpIr, ClampOpIr, ExpandOpIr, FlipOpIr,
-    FloatOperationIr, GatherOpIr, HandleContainer, InitOperationIr, MaskFillOpIr, MaskWhereOpIr,
-    NumericOperationIr, OperationIr, PermuteOpIr, RandomOpIr, ReduceDimOpIr,
-    ReduceDimWithIndicesOpIr, RepeatDimOpIr, ScalarOpIr, ScatterOpIr, SelectAssignOpIr, SelectOpIr,
-    SliceAssignOpIr, SliceOpIr, SwapDimsOpIr, TensorIr, UnaryOpIr,
+    BaseOperationIr, BinaryOpIr, CatOpIr, ClampOpIr, ExpandOpIr, FlipOpIr, FloatOperationIr,
+    GatherOpIr, HandleContainer, InitOperationIr, MaskFillOpIr, MaskWhereOpIr, NumericOperationIr,
+    OperationIr, PermuteOpIr, RandomOpIr, ReduceDimOpIr, ReduceDimWithIndicesOpIr, RepeatDimOpIr,
+    ScalarOpIr, ScatterOpIr, SelectAssignOpIr, SelectOpIr, SliceAssignOpIr, SliceOpIr,
+    SwapDimsOpIr, TensorIr, UnaryOpIr,
 };
 use burn_tensor::{
     Device, Distribution, Element, ElementConversion, Shape, TensorData, TensorMetadata,

crates/burn-fusion/src/ops/int.rs

@@ -6,7 +6,7 @@ use crate::{
     unary_int_ops,
 };
 use burn_ir::{
-    BackendIr, BaseOperationIr, BinaryOpIr, CatOpIr, ClampOpIr, ExpandOpIr, FlipOpIr, GatherOpIr,
+    BaseOperationIr, BinaryOpIr, CatOpIr, ClampOpIr, ExpandOpIr, FlipOpIr, GatherOpIr,
     HandleContainer, InitOperationIr, IntOperationIr, MaskFillOpIr, MaskWhereOpIr,
     NumericOperationIr, OperationIr, PermuteOpIr, RandomOpIr, ReduceDimOpIr,
     ReduceDimWithIndicesOpIr, RepeatDimOpIr, ScalarOpIr, ScatterOpIr, SelectAssignOpIr, SelectOpIr,

crates/burn-ndarray/src/ops/module.rs

@@ -109,8 +109,8 @@ impl<E: FloatNdArrayElement, I: IntNdArrayElement, Q: QuantElement> ModuleOps<Se
                     x.into(),
                     offset.into(),
                     weight.into(),
-                    mask.map(std::convert::Into::into),
-                    bias.map(std::convert::Into::into),
+                    mask.map(Into::into),
+                    bias.map(Into::into),
                 )
             }
         )

crates/burn-ndarray/src/ops/simd/cmp.rs

@@ -230,8 +230,8 @@ mod elemwise {
     use macerator::vload;
 
     use super::{
-        ArrayD, NdArrayElement, NdArrayTensor, PhantomData, Scalar, Simd, SimdCmpOp,
-        TensorMetadata, Vector, is_accelerated, seq, should_use_simd, vload_unaligned,
+        ArrayD, NdArrayElement, NdArrayTensor, PhantomData, Scalar, Simd, SimdCmpOp, Vector,
+        is_accelerated, seq, should_use_simd, vload_unaligned,
     };
 
     pub fn try_cmp_scalar_simd<E: NdArrayElement, T: NdArrayElement + Scalar, Op: SimdCmpOp<T>>(

crates/onnx-ir/src/node/pad.rs

@@ -95,7 +95,7 @@ pub fn pad_config(node: &Node) -> PadConfig {
 
         for (index, &item) in pads.iter().enumerate() {
             assert!(
-                !(!index_list.contains(&index) && item != 0),
+                index_list.contains(&index) || item == 0,
                 "Pad: padding will only be applied to the last two dimensions but found non zero padding for other dimensions"
             );
         }

crates/onnx-ir/src/node/reduce_max.rs

@@ -32,7 +32,7 @@ pub fn reduce_max_config(node: &Node) -> Option<usize> {
 
     // Not supported in Burn
     assert!(
-        !(!axes.is_empty() && keepdims == 0),
+        axes.is_empty() || keepdims != 0,
         "ReduceMax: the reduce operation must preserve the reduced dimension"
     );
 

crates/onnx-ir/src/node/reduce_mean.rs

@@ -32,7 +32,7 @@ pub fn reduce_mean_config(node: &Node) -> Option<usize> {
 
     // Not supported in Burn
     assert!(
-        !(!axes.is_empty() && keepdims == 0),
+        axes.is_empty() || keepdims != 0,
         "ReduceMean: the reduce operation must preserve the reduced dimension"
     );
 

crates/onnx-ir/src/node/reduce_min.rs

@@ -31,7 +31,7 @@ pub fn reduce_min_config(node: &Node) -> Option<usize> {
     );
 
     assert!(
-        !(!axes.is_empty() && keepdims == 0),
+        axes.is_empty() || keepdims != 0,
         "ReduceMin: the reduce operation must preserve the reduced dimension"
     );
 

crates/onnx-ir/src/node/reduce_prod.rs

@@ -33,7 +33,7 @@ pub fn reduce_prod_config(node: &Node) -> Option<usize> {
 
     // Not supported in Burn
     assert!(
-        !(!axes.is_empty() && keepdims == 0),
+        axes.is_empty() || keepdims != 0,
         "ReduceProd: the reduce operation must preserve the reduced dimension"
     );
 

crates/onnx-ir/src/node/reduce_sum.rs

@@ -42,7 +42,7 @@ pub fn reduce_sum_config(node: &Node) -> Option<usize> {
 
     // Not supported in Burn
     assert!(
-        !(!axes.is_empty() && keepdims == 0),
+        axes.is_empty() || keepdims != 0,
         "ReduceSum: the reduce operation must preserve the reduced dimension"
     );