xref: /llvm-project/mlir/test/Examples/Toy/Ch6/llvm-lowering.mlir (revision ee2c6cd9069fe0d8e7386ce53300e7645e4db792)
10372eb41SRiver Riddle// RUN: toyc-ch6 %s -emit=llvm -opt
20372eb41SRiver Riddle
3*ee2c6cd9SRiver Riddletoy.func @main() {
40050e8f0SRiver Riddle  %0 = toy.constant dense<[[1.000000e+00, 2.000000e+00, 3.000000e+00], [4.000000e+00, 5.000000e+00, 6.000000e+00]]> : tensor<2x3xf64>
50050e8f0SRiver Riddle  %2 = toy.transpose(%0 : tensor<2x3xf64>) to tensor<3x2xf64>
60050e8f0SRiver Riddle  %3 = toy.mul %2, %2 : tensor<3x2xf64>
70050e8f0SRiver Riddle  toy.print %3 : tensor<3x2xf64>
80050e8f0SRiver Riddle  toy.return
90372eb41SRiver Riddle}
100372eb41SRiver Riddle
110372eb41SRiver Riddle// CHECK-LABEL: define void @main()
120372eb41SRiver Riddle// CHECK: @printf
130372eb41SRiver Riddle// CHECK-SAME: 1.000000e+00
140372eb41SRiver Riddle// CHECK: @printf
150372eb41SRiver Riddle// CHECK-SAME: 1.600000e+01
160372eb41SRiver Riddle// CHECK: @printf
170372eb41SRiver Riddle// CHECK-SAME: 4.000000e+00
180372eb41SRiver Riddle// CHECK: @printf
190372eb41SRiver Riddle// CHECK-SAME: 2.500000e+01
200372eb41SRiver Riddle// CHECK: @printf
210372eb41SRiver Riddle// CHECK-SAME: 9.000000e+00
220372eb41SRiver Riddle// CHECK: @printf
230372eb41SRiver Riddle// CHECK-SAME: 3.000000e+01
24