# RUN: python -m toy %s --emit=toy --ir | filecheck %s

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

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