xref: /llvm-project/mlir/test/Examples/Toy/Ch5/scalar.toy (revision 0050e8f0cf5782217ebd78fa2b58be3aa9f8d9e2)
1c39592b0SMehdi Amini# RUN: toyc-ch5 %s -emit=mlir 2>&1 | FileCheck %s
2c39592b0SMehdi Amini
3c39592b0SMehdi Aminidef main() {
4c39592b0SMehdi Amini  var a<2, 2> = 5.5;
5c39592b0SMehdi Amini  print(a);
6c39592b0SMehdi Amini}
7c39592b0SMehdi Amini
8c39592b0SMehdi Amini# CHECK-LABEL: func @main() {
9*0050e8f0SRiver Riddle# CHECK-NEXT:    %0 = toy.constant dense<5.500000e+00> : tensor<f64>
10*0050e8f0SRiver Riddle# CHECK-NEXT:    %1 = toy.reshape(%0 : tensor<f64>) to tensor<2x2xf64>
11*0050e8f0SRiver Riddle# CHECK-NEXT:    toy.print %1 : tensor<2x2xf64>
12*0050e8f0SRiver Riddle# CHECK-NEXT:    toy.return
13c39592b0SMehdi Amini# CHECK-NEXT:  }
14c39592b0SMehdi Amini
15