forked from OSchip/llvm-project
15 lines
428 B
Plaintext
15 lines
428 B
Plaintext
# RUN: toyc-ch3 %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"() {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: }
|
|
|