# RUN: toyc-ch2 %s -emit=mlir 2>&1 | FileCheck %s | |
def main() { | |
var a<2, 2> = 5.5; | |
print(a); | |
} | |
# CHECK-LABEL: 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: } | |