File tree Expand file tree Collapse file tree 2 files changed +5
-7
lines changed
Expand file tree Collapse file tree 2 files changed +5
-7
lines changed Original file line number Diff line number Diff line change @@ -5777,11 +5777,9 @@ class ShapeCastCreateMaskFolderTrailingOneDim final
57775777};
57785778
57795779// / Pattern to rewrite Y = ShapeCast(Broadcast(X)) as either
5780- // /
5781- // / 1) Y = ShapeCast(X), or
5782- // / 2) Y = Broadcast(X)
5783- // /
5784- // / If both (1) and (2) are possible, (1) is chosen.
5780+ // / i) Y = ShapeCast(X), or
5781+ // / ii) Y = Broadcast(X)
5782+ // / If both (i) and (ii) are possible, (i) is chosen.
57855783class ShapeCastBroadcastFolder final : public OpRewritePattern<ShapeCastOp> {
57865784public:
57875785 using OpRewritePattern::OpRewritePattern;
Original file line number Diff line number Diff line change @@ -1006,10 +1006,10 @@ func.func @canonicalize_broadcast_shapecast_to_broadcast(%arg0: vector<3xf32>) -
10061006
10071007// -----
10081008
1009- // CHECK-LABEL: func @canonicalize_broadcast_ones_shapecast_to_broadcast_ones
1009+ // CHECK-LABEL: func @canonicalize_broadcast_shapecast_to_broadcast_ones
10101010// CHECK: vector.broadcast {{.*}} vector<1x1xi8> to vector<1x1x6x1x4xi8>
10111011// CHECK-NOT: vector.shape_cast
1012- func.func @canonicalize_broadcast_ones_shapecast_to_broadcast_ones (%arg0: vector <1 x1 xi8 >) -> vector <1 x1 x6 x1 x4 xi8 > {
1012+ func.func @canonicalize_broadcast_shapecast_to_broadcast_ones (%arg0: vector <1 x1 xi8 >) -> vector <1 x1 x6 x1 x4 xi8 > {
10131013 %0 = vector.broadcast %arg0 : vector <1 x1 xi8 > to vector <6 x4 xi8 >
10141014 %1 = vector.shape_cast %0 : vector <6 x4 xi8 > to vector <1 x1 x6 x1 x4 xi8 >
10151015 return %1 : vector <1 x1 x6 x1 x4 xi8 >
You can’t perform that action at this time.
0 commit comments