xref: /llvm-project/mlir/test/Examples/Toy/Ch6/llvm-lowering.mlir (revision ee2c6cd9069fe0d8e7386ce53300e7645e4db792) 
1// RUN: toyc-ch6 %s -emit=llvm -opt
2
3toy.func @main() {
4  %0 = toy.constant dense<[[1.000000e+00, 2.000000e+00, 3.000000e+00], [4.000000e+00, 5.000000e+00, 6.000000e+00]]> : tensor<2x3xf64>
5  %2 = toy.transpose(%0 : tensor<2x3xf64>) to tensor<3x2xf64>
6  %3 = toy.mul %2, %2 : tensor<3x2xf64>
7  toy.print %3 : tensor<3x2xf64>
8  toy.return
9}
10
11// CHECK-LABEL: define void @main()
12// CHECK: @printf
13// CHECK-SAME: 1.000000e+00
14// CHECK: @printf
15// CHECK-SAME: 1.600000e+01
16// CHECK: @printf
17// CHECK-SAME: 4.000000e+00
18// CHECK: @printf
19// CHECK-SAME: 2.500000e+01
20// CHECK: @printf
21// CHECK-SAME: 9.000000e+00
22// CHECK: @printf
23// CHECK-SAME: 3.000000e+01
24