From c37ac41c1afec9b3db7abeb9d65935fc94fa4077 Mon Sep 17 00:00:00 2001 From: Quentin Bourgerie Date: Thu, 9 Sep 2021 20:56:59 +0200 Subject: [PATCH] test(compiler): LowLFHEUnpametrize --- compiler/tests/Conversion/LowLFHEUnparametrize/func.mlir | 7 +++++++ .../LowLFHEUnparametrize/unrealized_conversion_cast.mlir | 8 ++++++++ 2 files changed, 15 insertions(+) create mode 100644 compiler/tests/Conversion/LowLFHEUnparametrize/func.mlir create mode 100644 compiler/tests/Conversion/LowLFHEUnparametrize/unrealized_conversion_cast.mlir diff --git a/compiler/tests/Conversion/LowLFHEUnparametrize/func.mlir b/compiler/tests/Conversion/LowLFHEUnparametrize/func.mlir new file mode 100644 index 000000000..852924948 --- /dev/null +++ b/compiler/tests/Conversion/LowLFHEUnparametrize/func.mlir @@ -0,0 +1,7 @@ +// RUN: zamacompiler --passes lowlfhe-unparametrize %s 2>&1| FileCheck %s + +// CHECK-LABEL: func @main(%arg0: !LowLFHE.lwe_ciphertext<_,_>) -> !LowLFHE.lwe_ciphertext<_,_> +func @main(%arg0: !LowLFHE.lwe_ciphertext<1024,4>) -> !LowLFHE.lwe_ciphertext<1024,4> { + // CHECK-NEXT: return %arg0 : !LowLFHE.lwe_ciphertext<_,_> + return %arg0: !LowLFHE.lwe_ciphertext<1024,4> +} \ No newline at end of file diff --git a/compiler/tests/Conversion/LowLFHEUnparametrize/unrealized_conversion_cast.mlir b/compiler/tests/Conversion/LowLFHEUnparametrize/unrealized_conversion_cast.mlir new file mode 100644 index 000000000..b48ba3c24 --- /dev/null +++ b/compiler/tests/Conversion/LowLFHEUnparametrize/unrealized_conversion_cast.mlir @@ -0,0 +1,8 @@ +// RUN: zamacompiler --passes lowlfhe-unparametrize %s 2>&1| FileCheck %s + +// CHECK-LABEL: func @main(%arg0: !LowLFHE.lwe_ciphertext<_,_>) -> !LowLFHE.lwe_ciphertext<_,_> +func @main(%arg0: !LowLFHE.lwe_ciphertext<1024,4>) -> !LowLFHE.lwe_ciphertext<_,_> { + // CHECK-NEXT: return %arg0 : !LowLFHE.lwe_ciphertext<_,_> + %0 = unrealized_conversion_cast %arg0 : !LowLFHE.lwe_ciphertext<1024,4> to !LowLFHE.lwe_ciphertext<_,_> + return %0: !LowLFHE.lwe_ciphertext<_,_> +} \ No newline at end of file