From e66fb20c4ff76eca17abc44cedaa99cc2c1d159f Mon Sep 17 00:00:00 2001 From: Quentin Bourgerie Date: Fri, 22 Oct 2021 15:36:25 +0200 Subject: [PATCH] test(compiler): Re introduce tests that was removed by a prior refacto --- 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..8374ce81e --- /dev/null +++ b/compiler/tests/Conversion/LowLFHEUnparametrize/func.mlir @@ -0,0 +1,7 @@ +// RUN: zamacompiler --passes lowlfhe-unparametrize --action=dump-llvm-dialect %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..f44a8404f --- /dev/null +++ b/compiler/tests/Conversion/LowLFHEUnparametrize/unrealized_conversion_cast.mlir @@ -0,0 +1,8 @@ +// RUN: zamacompiler --passes lowlfhe-unparametrize --action=dump-llvm-dialect %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 = builtin.unrealized_conversion_cast %arg0 : !LowLFHE.lwe_ciphertext<1024,4> to !LowLFHE.lwe_ciphertext<_,_> + return %0: !LowLFHE.lwe_ciphertext<_,_> +} \ No newline at end of file