diff --git a/compiler/tests/Conversion/HLFHELinalgToLinalg/zero.mlir b/compiler/tests/Conversion/HLFHELinalgToLinalg/zero.mlir new file mode 100644 index 000000000..642785d6d --- /dev/null +++ b/compiler/tests/Conversion/HLFHELinalgToLinalg/zero.mlir @@ -0,0 +1,14 @@ +// RUN: zamacompiler %s --action=dump-midlfhe --passes hlfhe-tensor-ops-to-linalg 2>&1 | FileCheck %s + +// CHECK: func @zero(%arg0: !HLFHE.eint<2>) -> tensor<3x2x!HLFHE.eint<2>> { +// CHECK-NEXT: %[[v0:.*]] = tensor.generate { +// CHECK-NEXT: ^bb0(%arg1: index, %arg2: index): // no predecessors +// CHECK-NEXT: %[[yld:.*]] = "HLFHE.zero"() : () -> !HLFHE.eint<2> +// CHECK-NEXT: tensor.yield %[[yld]] : !HLFHE.eint<2> +// CHECK-NEXT: } : tensor<3x2x!HLFHE.eint<2>> +// CHECK-NEXT: return %[[v0]] : tensor<3x2x!HLFHE.eint<2>> +// CHECK-NEXT: } +func @zero(%arg0: !HLFHE.eint<2>) -> tensor<3x2x!HLFHE.eint<2>> { + %1 = "HLFHELinalg.zero"(): () -> tensor<3x2x!HLFHE.eint<2>> + return %1 : tensor<3x2x!HLFHE.eint<2>> +}