struct-opt.mlir
706 Bytes
// 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]])