From ea3c940f4a79d5df823859aa04dab2dabcfc1ffd Mon Sep 17 00:00:00 2001
From: youben11 <ayoub.benaissa@zama.ai>
Date: Fri, 27 Aug 2021 10:21:04 +0100
Subject: [PATCH] feat: lower LowLFHE types (apply_lut) to llvm.ptr

---
 .../MLIRLowerableDialectsToLLVM.cpp                        | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/compiler/lib/Conversion/MLIRLowerableDialectsToLLVM/MLIRLowerableDialectsToLLVM.cpp b/compiler/lib/Conversion/MLIRLowerableDialectsToLLVM/MLIRLowerableDialectsToLLVM.cpp
index cc6340446..34ed4138e 100644
--- a/compiler/lib/Conversion/MLIRLowerableDialectsToLLVM/MLIRLowerableDialectsToLLVM.cpp
+++ b/compiler/lib/Conversion/MLIRLowerableDialectsToLLVM/MLIRLowerableDialectsToLLVM.cpp
@@ -64,7 +64,12 @@ void MLIRLowerableDialectsToLLVMPass::runOnOperation() {
 
 llvm::Optional<mlir::Type>
 MLIRLowerableDialectsToLLVMPass::convertTypes(mlir::Type type) {
-  if (type.isa<mlir::zamalang::LowLFHE::LweCiphertextType>()) {
+  if (type.isa<mlir::zamalang::LowLFHE::LweCiphertextType>() ||
+      type.isa<mlir::zamalang::LowLFHE::GlweCiphertextType>() ||
+      type.isa<mlir::zamalang::LowLFHE::LweKeySwitchKeyType>() ||
+      type.isa<mlir::zamalang::LowLFHE::LweBootstrapKeyType>() ||
+      type.isa<mlir::zamalang::LowLFHE::ForeignPlaintextListType>() ||
+      type.isa<mlir::zamalang::LowLFHE::PlaintextListType>()) {
     return mlir::LLVM::LLVMPointerType::get(
         mlir::IntegerType::get(type.getContext(), 64));
   }