rewrite test_silu.py

Author: GuoningHuang

tests/Python/test_silu.py

@@ -32,19 +32,17 @@ def foo(x):
 graph.lower_to_top_level_ir()
 print(graph._imported_module)
 
-# CHECK: module {
+#       CHECK: module {
 # CHECK-LABEL: func.func @forward
-# CHECK: %{{.*}} = tensor.empty() : tensor<4x4xf32>
-# CHECK: %[[RES:.*]] = linalg.generic {indexing_maps = [#map, #map], iterator_types = ["parallel", "parallel"]} ins(%arg0 : tensor<4x4xf32>) outs(%0 : tensor<4x4xf32>) {
-# CHECK: ^bb0(%in: f32, %out: f32):
-# CHECK: %{{.*}} = arith.negf %in : f32
-# CHECK: %{{.*}} = math.exp %{{.*}} : f32
-# CHECK: %{{.*}} = arith.constant 1.000000e+00 : f32
-# CHECK: %{{.*}} = arith.addf %{{.*}}, %{{.*}} : f32
-# CHECK: %{{.*}} = arith.divf %in, %{{.*}} : f32
-# CHECK: linalg.yield %{{.*}} : f32
-# CHECK: } -> tensor<4x4xf32>
-# CHECK: return %[[RES]] : tensor<4x4xf32>
-# CHECK: }
-# CHECK: }
+#       CHECK:   %[[EMPTY:.*]] = tensor.empty() : tensor<4x4xf32>
+#       CHECK:   %[[RES:.*]] = linalg.generic {.*} ins(%arg0 : tensor<4x4xf32>) outs(%[[EMPTY]] : tensor<4x4xf32>) {
+#       CHECK:   ^bb0(%in: f32, %out: f32):
+#       CHECK:       %[[NEG:.*]] = arith.negf %in : f32 
+#       CHECK:       %[[EXP:.*]] = math.exp %[[NEG]] : f32
+#       CHECK:       %[[ONE:.*]] = arith.constant 1.000000e+00 : f32
+#       CHECK:       %[[ADD:.*]] = arith.addf %[[EXP]], %[[ONE]] : f32
+#       CHECK:       %[[DIV:.*]] = arith.divf %in, %[[ADD]] : f32
+#       CHECK:       linalg.yield %[[DIV]] : f32
+#       CHECK:    } -> tensor<4x4xf32>
+#       CHECK: return %[[RES]] : tensor<4x4xf32>