diff options
Diffstat (limited to 'mlir/test/Examples/Toy/Ch7/struct-opt.mlir')
| -rw-r--r-- | mlir/test/Examples/Toy/Ch7/struct-opt.mlir | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/mlir/test/Examples/Toy/Ch7/struct-opt.mlir b/mlir/test/Examples/Toy/Ch7/struct-opt.mlir new file mode 100644 index 00000000000..8c4b055b4bf --- /dev/null +++ b/mlir/test/Examples/Toy/Ch7/struct-opt.mlir @@ -0,0 +1,16 @@ +// RUN: toyc-ch7 %s -emit=mlir -opt 2>&1 | FileCheck %s + +func @main() { + %0 = "toy.struct_constant"() { + value = [[dense<4.000000e+00> : tensor<2x2xf64>], dense<4.000000e+00> : tensor<2x2xf64>] + } : () -> !toy.struct<!toy.struct<tensor<*xf64>>, tensor<*xf64>> + %1 = "toy.struct_access"(%0) {index = 0 : i64} : (!toy.struct<!toy.struct<tensor<*xf64>>, tensor<*xf64>>) -> !toy.struct<tensor<*xf64>> + %2 = "toy.struct_access"(%1) {index = 0 : i64} : (!toy.struct<tensor<*xf64>>) -> tensor<*xf64> + "toy.print"(%2) : (tensor<*xf64>) -> () + "toy.return"() : () -> () +} + +// CHECK-LABEL: func @main +// CHECK-NEXT: %[[CST:.*]] = "toy.constant" +// CHECK-SAME: dense<4.0 +// CHECK-NEXT: "toy.print"(%[[CST]]) |

