Skip to content

Commit 89add3d

Browse files
Max191github-actions[bot]
authored andcommitted
check fusion in quantized matmul reassociation lit test
1 parent 085172d commit 89add3d

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

compiler/src/iree/compiler/GlobalOptimization/test/fuse_dequantization_matmul.mlir

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,7 @@ module {
9797
// REASSOCIATE-CHECK: %[[INITMATMUL:.+]] = tensor.empty() : tensor<11008x32xi32>
9898
// REASSOCIATE-CHECK: %[[FILLMATMUL:.+]] = linalg.fill ins(%[[C0I32]]
9999
// REASSOCIATE-CHECK-SAME: outs(%[[INITMATMUL]] :
100+
// REASSOCIATE-CHECK: %[[DISP:.+]] = flow.dispatch.region
100101
// REASSOCIATE-CHECK: %[[GENMATMUL:.+]] = linalg.generic
101102
// REASSOCIATE-CHECK-SAME: indexing_maps = [#[[MAP3]], #[[MAP4]], #[[MAP5]]]
102103
// REASSOCIATE-CHECK-SAME: iterator_types = ["parallel", "parallel", "reduction"]
@@ -122,4 +123,5 @@ module {
122123
// REASSOCIATE-CHECK: %[[RESUBF:.+]] = arith.subf %[[REMULF1]], %[[REMULF3]] : f32
123124
// REASSOCIATE-CHECK: %[[READDF:.+]] = arith.addf %[[RESUBF]], %[[REOUT0]] : f32
124125
// REASSOCIATE-CHECK: linalg.yield %[[READDF]] : f32
125-
// REASSOCIATE-CHECK: return %[[GENREASSOCIATE]]
126+
// REASSOCIATE-CHECK: flow.return %[[GENREASSOCIATE]]
127+
// REASSOCIATE-CHECK: return %[[DISP]]

0 commit comments

Comments
 (0)