diff options
Diffstat (limited to 'mlir/test/Examples/Toy/Ch6/llvm-lowering.mlir')
| -rw-r--r-- | mlir/test/Examples/Toy/Ch6/llvm-lowering.mlir | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/mlir/test/Examples/Toy/Ch6/llvm-lowering.mlir b/mlir/test/Examples/Toy/Ch6/llvm-lowering.mlir new file mode 100644 index 00000000000..12b050c3bfe --- /dev/null +++ b/mlir/test/Examples/Toy/Ch6/llvm-lowering.mlir @@ -0,0 +1,23 @@ +// RUN: toyc-ch6 %s -emit=llvm -opt + +func @main() { + %0 = "toy.constant"() {value = dense<[[1.000000e+00, 2.000000e+00, 3.000000e+00], [4.000000e+00, 5.000000e+00, 6.000000e+00]]> : tensor<2x3xf64>} : () -> tensor<2x3xf64> + %2 = "toy.transpose"(%0) : (tensor<2x3xf64>) -> tensor<3x2xf64> + %3 = "toy.mul"(%2, %2) : (tensor<3x2xf64>, tensor<3x2xf64>) -> tensor<3x2xf64> + "toy.print"(%3) : (tensor<3x2xf64>) -> () + "toy.return"() : () -> () +} + +// CHECK-LABEL: define void @main() +// CHECK: @printf +// CHECK-SAME: 1.000000e+00 +// CHECK: @printf +// CHECK-SAME: 1.600000e+01 +// CHECK: @printf +// CHECK-SAME: 4.000000e+00 +// CHECK: @printf +// CHECK-SAME: 2.500000e+01 +// CHECK: @printf +// CHECK-SAME: 9.000000e+00 +// CHECK: @printf +// CHECK-SAME: 3.000000e+01 |
