xref: /llvm-project/mlir/test/Examples/Toy/Ch4/scalar.toy (revision ee2c6cd9069fe0d8e7386ce53300e7645e4db792)
1d33a9dccSMehdi Amini# RUN: toyc-ch4 %s -emit=mlir 2>&1 | FileCheck %s
2d33a9dccSMehdi Amini
3d33a9dccSMehdi Aminidef main() {
4d33a9dccSMehdi Amini  var a<2, 2> = 5.5;
5d33a9dccSMehdi Amini  print(a);
6d33a9dccSMehdi Amini}
7d33a9dccSMehdi Amini
8*ee2c6cd9SRiver Riddle# CHECK-LABEL: toy.func @main() {
90050e8f0SRiver Riddle# CHECK-NEXT:    %0 = toy.constant dense<5.500000e+00> : tensor<f64>
100050e8f0SRiver Riddle# CHECK-NEXT:    %1 = toy.reshape(%0 : tensor<f64>) to tensor<2x2xf64>
110050e8f0SRiver Riddle# CHECK-NEXT:    toy.print %1 : tensor<2x2xf64>
120050e8f0SRiver Riddle# CHECK-NEXT:    toy.return
13d33a9dccSMehdi Amini# CHECK-NEXT:  }
14d33a9dccSMehdi Amini
15