aboutsummaryrefslogtreecommitdiff
path: root/mlir/test/Examples/Toy/Ch7/struct-opt.mlir
blob: 3faae88d2256612d1830b48e2fee77fac6cff208 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
// RUN: toyc-ch7 %s -emit=mlir -opt 2>&1 | FileCheck %s

toy.func @main() {
  %0 = toy.struct_constant [
    [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[0] : !toy.struct<!toy.struct<tensor<*xf64>>, tensor<*xf64>> -> !toy.struct<tensor<*xf64>>
  %2 = toy.struct_access %1[0] : !toy.struct<tensor<*xf64>> -> tensor<*xf64>
  toy.print %2 : tensor<*xf64>
  toy.return
}

// CHECK-LABEL: toy.func @main
// CHECK-NEXT: %[[CST:.*]] = toy.constant dense<4.0
// CHECK-NEXT: toy.print %[[CST]]