summaryrefslogtreecommitdiffstats
path: root/mlir/test/Examples/Toy/Ch7/struct-opt.mlir
diff options
context:
space:
mode:
Diffstat (limited to 'mlir/test/Examples/Toy/Ch7/struct-opt.mlir')
-rw-r--r--mlir/test/Examples/Toy/Ch7/struct-opt.mlir16
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]])
OpenPOWER on IntegriCloud