From 70bfff7f93be08835b2cf832e3b2bace1b321da9 Mon Sep 17 00:00:00 2001 From: Max Dawkins Date: Fri, 1 Dec 2023 16:30:24 -0500 Subject: [PATCH] check fusion in quantized matmul reassociation lit test --- .../GlobalOptimization/test/fuse_dequantization_matmul.mlir | 4 +++- 1 file changed, 3 insertions(+), 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 index fd8d196bcebe..e535c52eafcb 100644 --- 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]]