mlir/test/Examples/Toy/Ch7/scalar.toy


1
2
3
4
5
6
7
8
9
10
11
12
13
14

# RUN: toyc-ch7 %s -emit=mlir 2>&1 | FileCheck %s

def main() {
  var a<2, 2> = 5.5;
  print(a);
}

# CHECK-LABEL: toy.func @main() {
# CHECK-NEXT:    %0 = toy.constant dense<5.500000e+00> : tensor<f64>
# CHECK-NEXT:    %1 = toy.reshape(%0 : tensor<f64>) to tensor<2x2xf64>
# CHECK-NEXT:    toy.print %1 : tensor<2x2xf64>
# CHECK-NEXT:    toy.return
# CHECK-NEXT:  }