@@ -444,10 +444,10 @@ int simplify_iteration_two_strides(const int nd,
444
444
(strides2[i1] < 0 ) ? -strides2[i1] : strides2[i1];
445
445
auto abs_str2_i2 =
446
446
(strides2[i2] < 0 ) ? -strides2[i2] : strides2[i2];
447
- return (abs_str1_i1 > abs_str1_i2 ) ||
448
- (abs_str1_i1 == abs_str1_i2 &&
449
- (abs_str2_i1 > abs_str2_i2 ||
450
- (abs_str2_i1 == abs_str2_i2 && shape[i1] > shape[i2])));
447
+ return (abs_str2_i1 > abs_str2_i2 ) ||
448
+ (abs_str2_i1 == abs_str2_i2 &&
449
+ (abs_str1_i1 > abs_str1_i2 ||
450
+ (abs_str1_i1 == abs_str1_i2 && shape[i1] > shape[i2])));
451
451
});
452
452
453
453
std::vector<ShapeTy> shape_w;
@@ -600,12 +600,12 @@ int simplify_iteration_three_strides(const int nd,
600
600
(strides3[i1] < 0 ) ? -strides3[i1] : strides3[i1];
601
601
auto abs_str3_i2 =
602
602
(strides3[i2] < 0 ) ? -strides3[i2] : strides3[i2];
603
- return (abs_str1_i1 > abs_str1_i2 ) ||
604
- ((abs_str1_i1 == abs_str1_i2 ) &&
603
+ return (abs_str3_i1 > abs_str3_i2 ) ||
604
+ ((abs_str3_i1 == abs_str3_i2 ) &&
605
605
((abs_str2_i1 > abs_str2_i2) ||
606
606
((abs_str2_i1 == abs_str2_i2) &&
607
- ((abs_str3_i1 > abs_str3_i2 ) ||
608
- ((abs_str3_i1 == abs_str3_i2 ) &&
607
+ ((abs_str1_i1 > abs_str1_i2 ) ||
608
+ ((abs_str1_i1 == abs_str1_i2 ) &&
609
609
(shape[i1] > shape[i2]))))));
610
610
});
611
611
@@ -768,14 +768,14 @@ int simplify_iteration_four_strides(const int nd,
768
768
(strides4[i1] < 0 ) ? -strides4[i1] : strides4[i1];
769
769
auto abs_str4_i2 =
770
770
(strides4[i2] < 0 ) ? -strides4[i2] : strides4[i2];
771
- return (abs_str1_i1 > abs_str1_i2 ) ||
772
- ((abs_str1_i1 == abs_str1_i2 ) &&
773
- ((abs_str2_i1 > abs_str2_i2 ) ||
774
- ((abs_str2_i1 == abs_str2_i2 ) &&
775
- ((abs_str3_i1 > abs_str3_i2 ) ||
776
- ((abs_str3_i1 == abs_str3_i2 ) &&
777
- ((abs_str4_i1 > abs_str4_i2 ) ||
778
- ((abs_str4_i1 == abs_str4_i2 ) &&
771
+ return (abs_str4_i1 > abs_str4_i2 ) ||
772
+ ((abs_str4_i1 == abs_str4_i2 ) &&
773
+ ((abs_str3_i1 > abs_str3_i2 ) ||
774
+ ((abs_str3_i1 == abs_str3_i2 ) &&
775
+ ((abs_str2_i1 > abs_str2_i2 ) ||
776
+ ((abs_str2_i1 == abs_str2_i2 ) &&
777
+ ((abs_str1_i1 > abs_str1_i2 ) ||
778
+ ((abs_str1_i1 == abs_str1_i2 ) &&
779
779
(shape[i1] > shape[i2]))))))));
780
780
});
781
781
0 commit comments