check fusion in quantized matmul reassociation lit test

nod-ai · Dec 5, 2023 · 70bfff7 · 70bfff7
1 parent d354793
commit 70bfff7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/compiler/src/iree/compiler/GlobalOptimization/test/fuse_dequantization_matmul.mlir b/compiler/src/iree/compiler/GlobalOptimization/test/fuse_dequantization_matmul.mlir
@@ -97,6 +97,7 @@ module {
 //       REASSOCIATE-CHECK:   %[[INITMATMUL:.+]] = tensor.empty() : tensor<11008x32xi32>
 //       REASSOCIATE-CHECK:   %[[FILLMATMUL:.+]] = linalg.fill ins(%[[C0I32]]
 //  REASSOCIATE-CHECK-SAME:       outs(%[[INITMATMUL]] :
+//       REASSOCIATE-CHECK:   %[[DISP:.+]] = flow.dispatch.region
 //       REASSOCIATE-CHECK:   %[[GENMATMUL:.+]] = linalg.generic
 //  REASSOCIATE-CHECK-SAME:       indexing_maps = [#[[MAP3]], #[[MAP4]], #[[MAP5]]]
 //  REASSOCIATE-CHECK-SAME:       iterator_types = ["parallel", "parallel", "reduction"]
@@ -122,4 +123,5 @@ module {
 //       REASSOCIATE-CHECK:   %[[RESUBF:.+]] = arith.subf %[[REMULF1]], %[[REMULF3]] : f32
 //       REASSOCIATE-CHECK:   %[[READDF:.+]] = arith.addf %[[RESUBF]], %[[REOUT0]] : f32
 //       REASSOCIATE-CHECK:   linalg.yield %[[READDF]] : f32
-//       REASSOCIATE-CHECK:   return %[[GENREASSOCIATE]]
+//       REASSOCIATE-CHECK:   flow.return %[[GENREASSOCIATE]] 
+//       REASSOCIATE-CHECK:   return %[[DISP]]