Skip to content

Commit be8fe8b

Browse files
committed
test: refine FileCheck to capture SSA value in SiLU fusion test
1 parent b6c8e40 commit be8fe8b

File tree

2 files changed

+9
-7
lines changed

2 files changed

+9
-7
lines changed

frontend/Python/graph/transform/fuse_ops.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,4 +187,5 @@ def simply_fuse(graph: Graph):
187187
new_op_group.append(op)
188188
graph.op_groups = {}
189189
graph.op_groups["subgraph0"] = new_op_group
190-
graph.group_map_device = {"subgraph0": device}
190+
graph.group_map_device = {"subgraph0": device}
191+

tests/Python/test_silu.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -35,15 +35,16 @@ def foo(x):
3535
# CHECK: module {
3636
# CHECK-LABEL: func.func @forward
3737
# CHECK: %{{.*}} = tensor.empty() : tensor<4x4xf32>
38-
# CHECK: %{{.*}} = linalg.generic {indexing_maps = [#map, #map], iterator_types = ["parallel", "parallel"]} ins(%arg0 : tensor<4x4xf32>) outs(%0 : tensor<4x4xf32>) {
38+
# CHECK: %[[RES:.*]] = linalg.generic {indexing_maps = [#map, #map], iterator_types = ["parallel", "parallel"]} ins(%arg0 : tensor<4x4xf32>) outs(%0 : tensor<4x4xf32>) {
3939
# CHECK: ^bb0(%in: f32, %out: f32):
4040
# CHECK: %{{.*}} = arith.negf %in : f32
41-
# CHECK: %{{.*}} = math.exp %2 : f32
41+
# CHECK: %{{.*}} = math.exp %{{.*}} : f32
4242
# CHECK: %{{.*}} = arith.constant 1.000000e+00 : f32
43-
# CHECK: %{{.*}} = arith.addf %4, %3 : f32
44-
# CHECK: %{{.*}} = arith.divf %in, %5 : f32
45-
# CHECK: linalg.yield %6 : f32
43+
# CHECK: %{{.*}} = arith.addf %{{.*}}, %{{.*}} : f32
44+
# CHECK: %{{.*}} = arith.divf %in, %{{.*}} : f32
45+
# CHECK: linalg.yield %{{.*}} : f32
4646
# CHECK: } -> tensor<4x4xf32>
47-
# CHECK: return %1 : tensor<4x4xf32>
47+
# CHECK: return %[[RES]] : tensor<4x4xf32>
4848
# CHECK: }
4949
# CHECK: }
50+

0 commit comments

Comments
 (0)