refactor(compiler): Introduce compilation pipeline with multiple entries / exits

This refactoring commit restructures the compilation pipeline of `zamacompiler`, such that it is possible to enter and exit the pipeline at different points, effectively defining the level of abstraction at the input and the required level of abstraction for the output. The entry point is specified using the `--entry-dialect` argument. Valid choices are: `--entry-dialect=hlfhe`: Source contains HLFHE operations `--entry-dialect=midlfhe`: Source contains MidLFHE operations `--entry-dialect=lowlfhe`: Source contains LowLFHE operations `--entry-dialect=std`: Source does not contain any FHE Operations `--entry-dialect=llvm`: Source is in LLVM dialect The exit point is defined by an action, specified using --action. `--action=roundtrip`: Parse the source file to in-memory representation and immediately dump as text without any processing `--action=dump-midlfhe`: Lower source to MidLFHE and dump result as text `--action=dump-lowlfhe`: Lower source to LowLFHE and dump result as text `--action=dump-std`: Lower source to only standard MLIR dialects (i.e., all FHE operations have already been lowered) `--action=dump-llvm-dialect`: Lower source to MLIR's LLVM dialect (i.e., the LLVM dialect, not LLVM IR) `--action=dump-llvm-ir`: Lower source to plain LLVM IR (i.e., not the LLVM dialect, but actual LLVM IR) `--action=dump-optimized-llvm-ir`: Lower source to plain LLVM IR (i.e., not the LLVM dialect, but actual LLVM IR), pass the result through the LLVM optimizer and print the result. `--action=dump-jit-invoke`: Execute the full lowering pipeline to optimized LLVM IR, JIT compile the result, invoke the function specified in `--jit-funcname` with the parameters from `--jit-args` and print the functions return value.
2026-04-17 03:00:54 -04:00 · 2021-09-17 10:45:53 +02:00
parent ddebedd1d6
commit 30374ebb2c
58 changed files with 1014 additions and 862 deletions
--- a/compiler/tests/Conversion/HLFHEToMidLFHE/add_eint.mlir
+++ b/compiler/tests/Conversion/HLFHEToMidLFHE/add_eint.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler %s --passes hlfhe-to-midlfhe 2>&1| FileCheck %s
+// RUN: zamacompiler %s --entry-dialect=hlfhe --action=dump-midlfhe 2>&1| FileCheck %s

 // CHECK-LABEL: func @add_eint(%arg0: !MidLFHE.glwe<{_,_,_}{7}>, %arg1: !MidLFHE.glwe<{_,_,_}{7}>) -> !MidLFHE.glwe<{_,_,_}{7}>
 func @add_eint(%arg0: !HLFHE.eint<7>, %arg1: !HLFHE.eint<7>) -> !HLFHE.eint<7> {
@@ -7,4 +7,4 @@ func @add_eint(%arg0: !HLFHE.eint<7>, %arg1: !HLFHE.eint<7>) -> !HLFHE.eint<7> {

  %1 = "HLFHE.add_eint"(%arg0, %arg1): (!HLFHE.eint<7>, !HLFHE.eint<7>) -> (!HLFHE.eint<7>)
  return %1: !HLFHE.eint<7>
-}
+}
--- a/compiler/tests/Conversion/HLFHEToMidLFHE/add_eint_int.mlir
+++ b/compiler/tests/Conversion/HLFHEToMidLFHE/add_eint_int.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler %s --passes hlfhe-to-midlfhe 2>&1| FileCheck %s
+// RUN: zamacompiler %s --entry-dialect=hlfhe --action=dump-midlfhe 2>&1| FileCheck %s

 // CHECK-LABEL: func @add_eint_int(%arg0: !MidLFHE.glwe<{_,_,_}{7}>) -> !MidLFHE.glwe<{_,_,_}{7}>
 func @add_eint_int(%arg0: !HLFHE.eint<7>) -> !HLFHE.eint<7> {
--- a/compiler/tests/Conversion/HLFHEToMidLFHE/apply_univariate.mlir
+++ b/compiler/tests/Conversion/HLFHEToMidLFHE/apply_univariate.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler %s --passes hlfhe-to-midlfhe 2>&1| FileCheck %s
+// RUN: zamacompiler %s --entry-dialect=hlfhe --action=dump-midlfhe 2>&1| FileCheck %s

 // CHECK-LABEL: func @apply_lookup_table(%arg0: !MidLFHE.glwe<{_,_,_}{2}>, %arg1: tensor<4xi2>) -> !MidLFHE.glwe<{_,_,_}{2}>
 func @apply_lookup_table(%arg0: !HLFHE.eint<2>, %arg1: tensor<4xi2>) -> !HLFHE.eint<2> {
@@ -7,4 +7,4 @@ func @apply_lookup_table(%arg0: !HLFHE.eint<2>, %arg1: tensor<4xi2>) -> !HLFHE.e

  %1 = "HLFHE.apply_lookup_table"(%arg0, %arg1): (!HLFHE.eint<2>, tensor<4xi2>) -> (!HLFHE.eint<2>)
  return %1: !HLFHE.eint<2>
-}
+}
--- a/compiler/tests/Conversion/HLFHEToMidLFHE/apply_univariate_cst.mlir
+++ b/compiler/tests/Conversion/HLFHEToMidLFHE/apply_univariate_cst.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler %s --passes hlfhe-to-midlfhe 2>&1| FileCheck %s
+// RUN: zamacompiler %s --entry-dialect=hlfhe --action=dump-midlfhe 2>&1| FileCheck %s

 // CHECK-LABEL: func @apply_lookup_table_cst(%arg0: !MidLFHE.glwe<{_,_,_}{7}>) -> !MidLFHE.glwe<{_,_,_}{7}>
 func @apply_lookup_table_cst(%arg0: !HLFHE.eint<7>) -> !HLFHE.eint<7> {
@@ -8,4 +8,4 @@ func @apply_lookup_table_cst(%arg0: !HLFHE.eint<7>) -> !HLFHE.eint<7> {
  %tlu = std.constant dense<[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127]> : tensor<128xi64>
  %1 = "HLFHE.apply_lookup_table"(%arg0, %tlu): (!HLFHE.eint<7>, tensor<128xi64>) -> (!HLFHE.eint<7>)
  return %1: !HLFHE.eint<7>
-}
+}
--- a/compiler/tests/Conversion/HLFHEToMidLFHE/linalg_generic.mlir
+++ b/compiler/tests/Conversion/HLFHEToMidLFHE/linalg_generic.mlir
@@ -1,19 +1,18 @@
-// RUN: zamacompiler %s --passes hlfhe-to-midlfhe 2>&1| FileCheck %s
-
-//CHECK: #map0 = affine_map<(d0) -> (d0)>
-//CHECK-NEXT: #map1 = affine_map<(d0) -> (0)>
-//CHECK-NEXT: module  {
-//CHECK-NEXT:   func @linalg_generic(%arg0: tensor<2x!MidLFHE.glwe<{_,_,_}{2}>>, %arg1: tensor<2xi3>, %arg2: tensor<1x!MidLFHE.glwe<{_,_,_}{2}>>) {
-//CHECK-NEXT:     %0 = linalg.generic {indexing_maps = [#map0, #map0, #map1], iterator_types = ["reduction"]} ins(%arg0, %arg1 : tensor<2x!MidLFHE.glwe<{_,_,_}{2}>>, tensor<2xi3>) outs(%arg2 : tensor<1x!MidLFHE.glwe<{_,_,_}{2}>>) {
-//CHECK-NEXT:     ^bb0(%arg3: !MidLFHE.glwe<{_,_,_}{2}>, %arg4: i3, %arg5: !MidLFHE.glwe<{_,_,_}{2}>):  // no predecessors
-//CHECK-NEXT:       %1 = "MidLFHE.mul_glwe_int"(%arg3, %arg4) : (!MidLFHE.glwe<{_,_,_}{2}>, i3) -> !MidLFHE.glwe<{_,_,_}{2}>
-//CHECK-NEXT:       %2 = "MidLFHE.add_glwe"(%1, %arg5) : (!MidLFHE.glwe<{_,_,_}{2}>, !MidLFHE.glwe<{_,_,_}{2}>) -> !MidLFHE.glwe<{_,_,_}{2}>
-//CHECK-NEXT:       linalg.yield %2 : !MidLFHE.glwe<{_,_,_}{2}>
-//CHECK-NEXT:     } -> tensor<1x!MidLFHE.glwe<{_,_,_}{2}>>
-//CHECK-NEXT:     return
-//CHECK-NEXT:   }
-//CHECK-NEXT: }
+// RUN: zamacompiler %s --entry-dialect=hlfhe --action=dump-midlfhe 2>&1| FileCheck %s

+// CHECK: #map0 = affine_map<(d0) -> (d0)>
+// CHECK-NEXT: #map1 = affine_map<(d0) -> (0)>
+// CHECK-NEXT: module  {
+// CHECK-NEXT:   func @linalg_generic(%arg0: tensor<2x!MidLFHE.glwe<{_,_,_}{2}>>, %arg1: tensor<2xi3>, %arg2: tensor<1x!MidLFHE.glwe<{_,_,_}{2}>>) {
+// CHECK-NEXT:     %0 = linalg.generic {indexing_maps = [#map0, #map0, #map1], iterator_types = ["reduction"]} ins(%arg0, %arg1 : tensor<2x!MidLFHE.glwe<{_,_,_}{2}>>, tensor<2xi3>) outs(%arg2 : tensor<1x!MidLFHE.glwe<{_,_,_}{2}>>) {
+// CHECK-NEXT:     ^bb0(%arg3: !MidLFHE.glwe<{_,_,_}{2}>, %arg4: i3, %arg5: !MidLFHE.glwe<{_,_,_}{2}>):  // no predecessors
+// CHECK-NEXT:       %1 = "MidLFHE.mul_glwe_int"(%arg3, %arg4) : (!MidLFHE.glwe<{_,_,_}{2}>, i3) -> !MidLFHE.glwe<{_,_,_}{2}>
+// CHECK-NEXT:       %2 = "MidLFHE.add_glwe"(%1, %arg5) : (!MidLFHE.glwe<{_,_,_}{2}>, !MidLFHE.glwe<{_,_,_}{2}>) -> !MidLFHE.glwe<{_,_,_}{2}>
+// CHECK-NEXT:       linalg.yield %2 : !MidLFHE.glwe<{_,_,_}{2}>
+// CHECK-NEXT:     } -> tensor<1x!MidLFHE.glwe<{_,_,_}{2}>>
+// CHECK-NEXT:     return
+// CHECK-NEXT:   }
+// CHECK-NEXT: }

 #map0 = affine_map<(d0) -> (d0)>
 #map1 = affine_map<(d0) -> (0)>
--- a/compiler/tests/Conversion/HLFHEToMidLFHE/mul_eint_int.mlir
+++ b/compiler/tests/Conversion/HLFHEToMidLFHE/mul_eint_int.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler %s --passes hlfhe-to-midlfhe 2>&1| FileCheck %s
+// RUN: zamacompiler %s --entry-dialect=hlfhe --action=dump-midlfhe 2>&1| FileCheck %s

 // CHECK-LABEL: func @mul_eint_int(%arg0: !MidLFHE.glwe<{_,_,_}{7}>) -> !MidLFHE.glwe<{_,_,_}{7}>
 func @mul_eint_int(%arg0: !HLFHE.eint<7>) -> !HLFHE.eint<7> {
@@ -9,4 +9,4 @@ func @mul_eint_int(%arg0: !HLFHE.eint<7>) -> !HLFHE.eint<7> {
  %0 = constant 1 : i8
  %1 = "HLFHE.mul_eint_int"(%arg0, %0): (!HLFHE.eint<7>, i8) -> (!HLFHE.eint<7>)
  return %1: !HLFHE.eint<7>
-}
+}
--- a/compiler/tests/Conversion/HLFHEToMidLFHE/sub_int_eint.mlir
+++ b/compiler/tests/Conversion/HLFHEToMidLFHE/sub_int_eint.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler %s --passes hlfhe-to-midlfhe 2>&1| FileCheck %s
+// RUN: zamacompiler %s --entry-dialect=hlfhe --action=dump-midlfhe 2>&1| FileCheck %s

 // CHECK-LABEL: func @sub_int_eint(%arg0: !MidLFHE.glwe<{_,_,_}{7}>) -> !MidLFHE.glwe<{_,_,_}{7}>
 func @sub_int_eint(%arg0: !HLFHE.eint<7>) -> !HLFHE.eint<7> {
--- a/compiler/tests/Conversion/LowLFHEToConcreteCAPI/bootstrap.mlir
+++ b/compiler/tests/Conversion/LowLFHEToConcreteCAPI/bootstrap.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --passes lowlfhe-to-concrete-c-api %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=lowlfhe --action=dump-std %s 2>&1| FileCheck %s

 // CHECK-LABEL: module
 // CHECK-NEXT: func private @add_plaintext_list_glwe_ciphertext_u64(index, !LowLFHE.glwe_ciphertext, !LowLFHE.glwe_ciphertext, !LowLFHE.plaintext_list)
@@ -27,4 +27,4 @@ func @bootstrap_lwe(%arg0: !LowLFHE.lwe_ciphertext<1024,4>, %arg1: !LowLFHE.glwe
  // CHECK-NEXT: return %[[RES]] : !LowLFHE.lwe_ciphertext<1024,4>
  %1 = "LowLFHE.bootstrap_lwe"(%arg0, %arg1) {baseLog = 2 : i32, k = 1 : i32, level = 3 : i32, polynomialSize = 1024 : i32} : (!LowLFHE.lwe_ciphertext<1024,4>, !LowLFHE.glwe_ciphertext) -> !LowLFHE.lwe_ciphertext<1024,4>
  return %1: !LowLFHE.lwe_ciphertext<1024,4>
-}
+}
--- a/compiler/tests/Conversion/LowLFHEToConcreteCAPI/glwe_from_table.mlir
+++ b/compiler/tests/Conversion/LowLFHEToConcreteCAPI/glwe_from_table.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --passes lowlfhe-to-concrete-c-api %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=lowlfhe --action=dump-std %s 2>&1| FileCheck %s

 // CHECK-LABEL: module
 // CHECK-NEXT: func private @runtime_foreign_plaintext_list_u64(index, tensor<16xi64>, i64, i32) -> !LowLFHE.foreign_plaintext_list
@@ -31,4 +31,4 @@ func @glwe_from_table(%arg0: tensor<16xi64>) -> !LowLFHE.glwe_ciphertext {
  // CHECK-NEXT: return %[[V1]] : !LowLFHE.glwe_ciphertext
  %1 = "LowLFHE.glwe_from_table"(%arg0) {k = 1 : i32, p = 4 : i32, polynomialSize = 1024 : i32} : (tensor<16xi64>) -> !LowLFHE.glwe_ciphertext
  return %1: !LowLFHE.glwe_ciphertext
-}
+}
--- a/compiler/tests/Conversion/LowLFHEToConcreteCAPI/keyswitch_lwe.mlir
+++ b/compiler/tests/Conversion/LowLFHEToConcreteCAPI/keyswitch_lwe.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --passes lowlfhe-to-concrete-c-api %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=lowlfhe --action=dump-std %s 2>&1| FileCheck %s

 // CHECK-LABEL: module
 // CHECK-NEXT: func private @add_plaintext_list_glwe_ciphertext_u64(index, !LowLFHE.glwe_ciphertext, !LowLFHE.glwe_ciphertext, !LowLFHE.plaintext_list)
@@ -26,4 +26,4 @@ func @keyswitch_lwe(%arg0: !LowLFHE.lwe_ciphertext<1024,4>) -> !LowLFHE.lwe_ciph
  // CHECK-NEXT: return %[[RES]] : !LowLFHE.lwe_ciphertext<1024,4>
  %1 = "LowLFHE.keyswitch_lwe"(%arg0) {baseLog = 2 : i32, inputLweSize = 1 : i32, level = 3 : i32, outputLweSize = 1 : i32} : (!LowLFHE.lwe_ciphertext<1024,4>) -> !LowLFHE.lwe_ciphertext<1024,4>
  return %1: !LowLFHE.lwe_ciphertext<1024,4>
-}
+}
--- a/compiler/tests/Conversion/LowLFHEUnparametrize/func.mlir
+++ b/compiler/tests/Conversion/LowLFHEUnparametrize/func.mlir
@@ -1,7 +0,0 @@
-// 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>
-}
--- a/compiler/tests/Conversion/LowLFHEUnparametrize/unrealized_conversion_cast.mlir
+++ b/compiler/tests/Conversion/LowLFHEUnparametrize/unrealized_conversion_cast.mlir
@@ -1,8 +0,0 @@
-// 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<_,_>
-}
--- a/compiler/tests/Conversion/MidLFHEToLowLFHE/add_glwe.mlir
+++ b/compiler/tests/Conversion/MidLFHEToLowLFHE/add_glwe.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --passes midlfhe-to-lowlfhe %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=midlfhe --action=dump-lowlfhe --parametrize-midlfhe=false %s 2>&1| FileCheck %s

 // CHECK-LABEL: func @add_glwe(%arg0: !LowLFHE.lwe_ciphertext<2048,7>, %arg1: !LowLFHE.lwe_ciphertext<2048,7>) -> !LowLFHE.lwe_ciphertext<2048,7>
 func @add_glwe(%arg0: !MidLFHE.glwe<{2048,1,64}{7}>, %arg1: !MidLFHE.glwe<{2048,1,64}{7}>) -> !MidLFHE.glwe<{2048,1,64}{7}> {
--- a/compiler/tests/Conversion/MidLFHEToLowLFHE/add_glwe_int.mlir
+++ b/compiler/tests/Conversion/MidLFHEToLowLFHE/add_glwe_int.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --passes midlfhe-to-lowlfhe %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=midlfhe --action=dump-lowlfhe --parametrize-midlfhe=false %s 2>&1| FileCheck %s

 // CHECK-LABEL: func @add_glwe_const_int(%arg0: !LowLFHE.lwe_ciphertext<1024,7>) -> !LowLFHE.lwe_ciphertext<1024,7>
 func @add_glwe_const_int(%arg0: !MidLFHE.glwe<{1024,1,64}{7}>) -> !MidLFHE.glwe<{1024,1,64}{7}> {
@@ -19,4 +19,4 @@ func @add_glwe_int(%arg0: !MidLFHE.glwe<{1024,1,64}{4}>, %arg1: i5) -> !MidLFHE.
  // CHECK-NEXT: return %[[V2]] : !LowLFHE.lwe_ciphertext<1024,4>
  %1 = "MidLFHE.add_glwe_int"(%arg0, %arg1): (!MidLFHE.glwe<{1024,1,64}{4}>, i5) -> (!MidLFHE.glwe<{1024,1,64}{4}>)
  return %1: !MidLFHE.glwe<{1024,1,64}{4}>
-}
+}
--- a/compiler/tests/Conversion/MidLFHEToLowLFHE/apply_lookup_table.mlir
+++ b/compiler/tests/Conversion/MidLFHEToLowLFHE/apply_lookup_table.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --passes midlfhe-to-lowlfhe %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=midlfhe --action=dump-lowlfhe --parametrize-midlfhe=false %s 2>&1| FileCheck %s

 // CHECK-LABEL: func @apply_lookup_table(%arg0: !LowLFHE.lwe_ciphertext<1024,4>, %arg1: tensor<16xi4>) -> !LowLFHE.lwe_ciphertext<1024,4>
 func @apply_lookup_table(%arg0: !MidLFHE.glwe<{1024,1,64}{4}>, %arg1: tensor<16xi4>) -> !MidLFHE.glwe<{1024,1,64}{4}> {
@@ -8,4 +8,4 @@ func @apply_lookup_table(%arg0: !MidLFHE.glwe<{1024,1,64}{4}>, %arg1: tensor<16x
  // CHECK-NEXT: return %[[V3]] : !LowLFHE.lwe_ciphertext<1024,4>
  %1 = "MidLFHE.apply_lookup_table"(%arg0, %arg1){k=1:i32, polynomialSize=1024:i32, levelKS=3:i32, baseLogKS=2:i32, levelBS=5:i32, baseLogBS=4:i32, outputSizeKS=600:i32}: (!MidLFHE.glwe<{1024,1,64}{4}>, tensor<16xi4>) -> (!MidLFHE.glwe<{1024,1,64}{4}>)
  return %1: !MidLFHE.glwe<{1024,1,64}{4}>
-}
+}
--- a/compiler/tests/Conversion/MidLFHEToLowLFHE/apply_lookup_table_cst.mlir
+++ b/compiler/tests/Conversion/MidLFHEToLowLFHE/apply_lookup_table_cst.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --passes midlfhe-to-lowlfhe %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=midlfhe --action=dump-lowlfhe --parametrize-midlfhe=false %s 2>&1| FileCheck %s

 // CHECK-LABEL: func @apply_lookup_table_cst(%arg0: !LowLFHE.lwe_ciphertext<2048,4>) -> !LowLFHE.lwe_ciphertext<2048,4>
 func @apply_lookup_table_cst(%arg0: !MidLFHE.glwe<{2048,1,64}{4}>) -> !MidLFHE.glwe<{2048,1,64}{4}> {
@@ -10,4 +10,4 @@ func @apply_lookup_table_cst(%arg0: !MidLFHE.glwe<{2048,1,64}{4}>) -> !MidLFHE.g
  %tlu = std.constant dense<[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]> : tensor<16xi4>
  %1 = "MidLFHE.apply_lookup_table"(%arg0, %tlu){k=1:i32, polynomialSize=2048:i32, levelKS=3:i32, baseLogKS=2:i32, levelBS=5:i32, baseLogBS=4:i32, outputSizeKS=600:i32}: (!MidLFHE.glwe<{2048,1,64}{4}>, tensor<16xi4>) -> (!MidLFHE.glwe<{2048,1,64}{4}>)
  return %1: !MidLFHE.glwe<{2048,1,64}{4}>
-}
+}
--- a/compiler/tests/Conversion/MidLFHEToLowLFHE/mul_glwe_int.mlir
+++ b/compiler/tests/Conversion/MidLFHEToLowLFHE/mul_glwe_int.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --passes midlfhe-to-lowlfhe %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=midlfhe --action=dump-lowlfhe --parametrize-midlfhe=false %s 2>&1| FileCheck %s

 // CHECK-LABEL: func @mul_glwe_const_int(%arg0: !LowLFHE.lwe_ciphertext<1024,7>) -> !LowLFHE.lwe_ciphertext<1024,7>
 func @mul_glwe_const_int(%arg0: !MidLFHE.glwe<{1024,1,64}{7}>) -> !MidLFHE.glwe<{1024,1,64}{7}> {
@@ -19,4 +19,4 @@ func @mul_glwe_int(%arg0: !MidLFHE.glwe<{1024,1,64}{4}>, %arg1: i5) -> !MidLFHE.
  // CHECK-NEXT: return %[[V2]] : !LowLFHE.lwe_ciphertext<1024,4>
  %1 = "MidLFHE.mul_glwe_int"(%arg0, %arg1): (!MidLFHE.glwe<{1024,1,64}{4}>, i5) -> (!MidLFHE.glwe<{1024,1,64}{4}>)
  return %1: !MidLFHE.glwe<{1024,1,64}{4}>
-}
+}
--- a/compiler/tests/Conversion/MidLFHEToLowLFHE/sub_int_glwe.mlir
+++ b/compiler/tests/Conversion/MidLFHEToLowLFHE/sub_int_glwe.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --passes midlfhe-to-lowlfhe %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=midlfhe --action=dump-lowlfhe --parametrize-midlfhe=false %s 2>&1| FileCheck %s

 // CHECK-LABEL: func @sub_const_int_glwe(%arg0: !LowLFHE.lwe_ciphertext<1024,7>) -> !LowLFHE.lwe_ciphertext<1024,7>
 func @sub_const_int_glwe(%arg0: !MidLFHE.glwe<{1024,1,64}{7}>) -> !MidLFHE.glwe<{1024,1,64}{7}> {
@@ -20,4 +20,4 @@ func @sub_int_glwe(%arg0: !MidLFHE.glwe<{1024,1,64}{4}>, %arg1: i5) -> !MidLFHE.
  // CHECK-NEXT: return %[[V2]] : !LowLFHE.lwe_ciphertext<1024,4>
  %1 = "MidLFHE.sub_int_glwe"(%arg1, %arg0): (i5, !MidLFHE.glwe<{1024,1,64}{4}>) -> (!MidLFHE.glwe<{1024,1,64}{4}>)
  return %1: !MidLFHE.glwe<{1024,1,64}{4}>
-}
+}
--- a/compiler/tests/Dialect/HLFHE/dot.invalid.mlir
+++ b/compiler/tests/Dialect/HLFHE/dot.invalid.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --split-input-file --verify-diagnostics %s
+// RUN: zamacompiler --split-input-file --verify-diagnostics --entry-dialect=hlfhe --action=roundtrip %s

 // Incompatible shapes
 func @dot_incompatible_shapes(
@@ -66,4 +66,4 @@ func @dot_incompatible_int(
    (tensor<4x!HLFHE.eint<2>>, tensor<4xi4>) -> !HLFHE.eint<2>

  return %ret : !HLFHE.eint<2>
-}
+}
--- a/compiler/tests/Dialect/HLFHE/eint_error_p_too_big.mlir
+++ b/compiler/tests/Dialect/HLFHE/eint_error_p_too_big.mlir
@@ -1,4 +1,4 @@
-// RUN: not zamacompiler %s  2>&1| FileCheck %s
+// RUN: not zamacompiler --entry-dialect=hlfhe --action=roundtrip %s  2>&1| FileCheck %s

 // CHECK-LABEL: eint support only precision in ]0;7]
 func @test(%arg0: !HLFHE.eint<8>) {
--- a/compiler/tests/Dialect/HLFHE/eint_error_p_too_small.mlir
+++ b/compiler/tests/Dialect/HLFHE/eint_error_p_too_small.mlir
@@ -1,4 +1,4 @@
-// RUN: not zamacompiler %s  2>&1| FileCheck %s
+// RUN: not zamacompiler --entry-dialect=hlfhe --action=roundtrip %s  2>&1| FileCheck %s

 // CHECK-LABEL: eint support only precision in ]0;7]
 func @test(%arg0: !HLFHE.eint<0>) {
--- a/compiler/tests/Dialect/HLFHE/op_add_eint_err_inputs.mlir
+++ b/compiler/tests/Dialect/HLFHE/op_add_eint_err_inputs.mlir
@@ -1,7 +1,7 @@
-// RUN: not zamacompiler %s 2>&1| FileCheck %s
+// RUN: not zamacompiler --entry-dialect=hlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: error: 'HLFHE.add_eint' op  should have the width of encrypted inputs equals
 func @add_eint(%arg0: !HLFHE.eint<2>, %arg1: !HLFHE.eint<3>) -> !HLFHE.eint<2> {
  %1 = "HLFHE.add_eint"(%arg0, %arg1): (!HLFHE.eint<2>, !HLFHE.eint<3>) -> (!HLFHE.eint<2>)
  return %1: !HLFHE.eint<2>
-}
+}
--- a/compiler/tests/Dialect/HLFHE/op_add_eint_err_result.mlir
+++ b/compiler/tests/Dialect/HLFHE/op_add_eint_err_result.mlir
@@ -1,7 +1,7 @@
-// RUN: not zamacompiler %s 2>&1| FileCheck %s
+// RUN: not zamacompiler --entry-dialect=hlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: error: 'HLFHE.add_eint' op  should have the width of encrypted inputs and result equals
 func @add_eint(%arg0: !HLFHE.eint<2>, %arg1: !HLFHE.eint<2>) -> !HLFHE.eint<3> {
  %1 = "HLFHE.add_eint"(%arg0, %arg1): (!HLFHE.eint<2>, !HLFHE.eint<2>) -> (!HLFHE.eint<3>)
  return %1: !HLFHE.eint<3>
-}
+}
--- a/compiler/tests/Dialect/HLFHE/op_add_eint_int_err_inputs.mlir
+++ b/compiler/tests/Dialect/HLFHE/op_add_eint_int_err_inputs.mlir
@@ -1,8 +1,8 @@
-// RUN: not zamacompiler %s 2>&1| FileCheck %s
+// RUN: not zamacompiler --entry-dialect=hlfhe --action=roundtrip  %s 2>&1| FileCheck %s

 // CHECK-LABEL: error: 'HLFHE.add_eint_int' op  should have the width of plain input equals to width of encrypted input + 1
 func @add_eint_int(%arg0: !HLFHE.eint<2>) -> !HLFHE.eint<2> {
  %0 = constant 1 : i4
  %1 = "HLFHE.add_eint_int"(%arg0, %0): (!HLFHE.eint<2>, i4) -> (!HLFHE.eint<2>)
  return %1: !HLFHE.eint<2>
-}
+}
--- a/compiler/tests/Dialect/HLFHE/op_add_eint_int_err_result.mlir
+++ b/compiler/tests/Dialect/HLFHE/op_add_eint_int_err_result.mlir
@@ -1,8 +1,8 @@
-// RUN: not zamacompiler %s 2>&1| FileCheck %s
+// RUN: not zamacompiler --entry-dialect=hlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: error: 'HLFHE.add_eint_int' op  should have the width of encrypted inputs and result equals
 func @add_eint_int(%arg0: !HLFHE.eint<2>) -> !HLFHE.eint<3> {
  %0 = constant 1 : i2
  %1 = "HLFHE.add_eint_int"(%arg0, %0): (!HLFHE.eint<2>, i2) -> (!HLFHE.eint<3>)
  return %1: !HLFHE.eint<3>
-}
+}
--- a/compiler/tests/Dialect/HLFHE/op_apply_lookup_table_bad_dimension.mlir
+++ b/compiler/tests/Dialect/HLFHE/op_apply_lookup_table_bad_dimension.mlir
@@ -1,7 +1,7 @@
-// RUN: not zamacompiler %s 2>&1| FileCheck %s
+// RUN: not zamacompiler --entry-dialect=hlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: error: 'HLFHE.apply_lookup_table' op  should have as `l_cst` argument a shape of one dimension equals to 2^p, where p is the width of the `ct` argument.
 func @apply_lookup_table(%arg0: !HLFHE.eint<2>, %arg1: tensor<8xi3>) -> !HLFHE.eint<2> {
  %1 = "HLFHE.apply_lookup_table"(%arg0, %arg1): (!HLFHE.eint<2>, tensor<8xi3>) -> (!HLFHE.eint<2>)
  return %1: !HLFHE.eint<2>
-}
+}
--- a/compiler/tests/Dialect/HLFHE/op_mul_eint_int_err_inputs.mlir
+++ b/compiler/tests/Dialect/HLFHE/op_mul_eint_int_err_inputs.mlir
@@ -1,8 +1,8 @@
-// RUN: not zamacompiler %s 2>&1| FileCheck %s
+// RUN: not zamacompiler --entry-dialect=hlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: error: 'HLFHE.mul_eint_int' op  should have the width of plain input equals to width of encrypted input + 1
 func @mul_eint_int(%arg0: !HLFHE.eint<2>) -> !HLFHE.eint<2> {
  %0 = constant 1 : i4
  %1 = "HLFHE.mul_eint_int"(%arg0, %0): (!HLFHE.eint<2>, i4) -> (!HLFHE.eint<2>)
  return %1: !HLFHE.eint<2>
-}
+}
--- a/compiler/tests/Dialect/HLFHE/op_mul_eint_int_err_result.mlir
+++ b/compiler/tests/Dialect/HLFHE/op_mul_eint_int_err_result.mlir
@@ -1,8 +1,8 @@
-// RUN: not zamacompiler %s 2>&1| FileCheck %s
+// RUN: not zamacompiler --entry-dialect=hlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: error: 'HLFHE.mul_eint_int' op  should have the width of encrypted inputs and result equals
 func @mul_eint_int(%arg0: !HLFHE.eint<2>) -> !HLFHE.eint<3> {
  %0 = constant 1 : i2
  %1 = "HLFHE.mul_eint_int"(%arg0, %0): (!HLFHE.eint<2>, i2) -> (!HLFHE.eint<3>)
  return %1: !HLFHE.eint<3>
-}
+}
--- a/compiler/tests/Dialect/HLFHE/op_sub_int_eint_err_inputs.mlir
+++ b/compiler/tests/Dialect/HLFHE/op_sub_int_eint_err_inputs.mlir
@@ -1,8 +1,8 @@
-// RUN: not zamacompiler %s 2>&1| FileCheck %s
+// RUN: not zamacompiler --entry-dialect=hlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: error: 'HLFHE.sub_int_eint' op  should have the width of plain input equals to width of encrypted input + 1
 func @sub_int_eint(%arg0: !HLFHE.eint<2>) -> !HLFHE.eint<2> {
  %0 = constant 1 : i4
  %1 = "HLFHE.sub_int_eint"(%0, %arg0): (i4, !HLFHE.eint<2>) -> (!HLFHE.eint<2>)
  return %1: !HLFHE.eint<2>
-}
+}
--- a/compiler/tests/Dialect/HLFHE/op_sub_int_eint_err_result.mlir
+++ b/compiler/tests/Dialect/HLFHE/op_sub_int_eint_err_result.mlir
@@ -1,8 +1,8 @@
-// RUN: not zamacompiler %s 2>&1| FileCheck %s
+// RUN: not zamacompiler --entry-dialect=hlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: error: 'HLFHE.sub_int_eint' op  should have the width of encrypted inputs and result equals
 func @sub_int_eint(%arg0: !HLFHE.eint<2>) -> !HLFHE.eint<3> {
  %0 = constant 1 : i2
  %1 = "HLFHE.sub_int_eint"(%0, %arg0): (i2, !HLFHE.eint<2>) -> (!HLFHE.eint<3>)
  return %1: !HLFHE.eint<3>
-}
+}
--- a/compiler/tests/Dialect/HLFHE/ops.mlir
+++ b/compiler/tests/Dialect/HLFHE/ops.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --round-trip %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=hlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: func @zero() -> !HLFHE.eint<2>
 func @zero() -> !HLFHE.eint<2> {
--- a/compiler/tests/Dialect/HLFHE/tensor-ops-to-linalg.mlir
+++ b/compiler/tests/Dialect/HLFHE/tensor-ops-to-linalg.mlir
@@ -1,22 +1,22 @@
-// RUN: zamacompiler %s --passes hlfhe-tensor-ops-to-linalg 2>&1 | FileCheck %s
+// RUN: zamacompiler %s --entry-dialect=hlfhe --action=dump-midlfhe 2>&1 | FileCheck %s

 //CHECK: #map0 = affine_map<(d0) -> (d0)>
 //CHECK-NEXT: #map1 = affine_map<(d0) -> (0)>
 //CHECK-NEXT: module  {
-//CHECK-NEXT:   func @dot_eint_int(%arg0: tensor<2x!HLFHE.eint<2>>, %arg1: tensor<2xi3>) -> !HLFHE.eint<2> {
-//CHECK-NEXT:     %0 = "HLFHE.zero"() : () -> !HLFHE.eint<2>
-//CHECK-NEXT:     %1 = tensor.from_elements %0 : tensor<1x!HLFHE.eint<2>>
-//CHECK-NEXT:     %2 = linalg.generic {indexing_maps = [#map0, #map0, #map1], iterator_types = ["reduction"]} ins(%arg0, %arg1 : tensor<2x!HLFHE.eint<2>>, tensor<2xi3>) outs(%1 : tensor<1x!HLFHE.eint<2>>) {
-//CHECK-NEXT:     ^bb0(%arg2: !HLFHE.eint<2>, %arg3: i3, %arg4: !HLFHE.eint<2>):  // no predecessors
-//CHECK-NEXT:       %4 = "HLFHE.mul_eint_int"(%arg2, %arg3) : (!HLFHE.eint<2>, i3) -> !HLFHE.eint<2>
-//CHECK-NEXT:       %5 = "HLFHE.add_eint"(%4, %arg4) : (!HLFHE.eint<2>, !HLFHE.eint<2>) -> !HLFHE.eint<2>
-//CHECK-NEXT:       linalg.yield %5 : !HLFHE.eint<2>
-//CHECK-NEXT:     } -> tensor<1x!HLFHE.eint<2>>
+//CHECK-NEXT:   func @dot_eint_int(%arg0: tensor<2x!MidLFHE.glwe<{_,_,_}{2}>>, %arg1: tensor<2xi3>) -> !MidLFHE.glwe<{_,_,_}{2}> {
+//CHECK-NEXT:     %0 = "MidLFHE.zero"() : () -> !MidLFHE.glwe<{_,_,_}{2}>
+//CHECK-NEXT:     %1 = tensor.from_elements %0 : tensor<1x!MidLFHE.glwe<{_,_,_}{2}>>
+//CHECK-NEXT:     %2 = linalg.generic {indexing_maps = [#map0, #map0, #map1], iterator_types = ["reduction"]} ins(%arg0, %arg1 : tensor<2x!MidLFHE.glwe<{_,_,_}{2}>>, tensor<2xi3>) outs(%1 : tensor<1x!MidLFHE.glwe<{_,_,_}{2}>>) {
+//CHECK-NEXT:     ^bb0(%arg2: !MidLFHE.glwe<{_,_,_}{2}>, %arg3: i3, %arg4: !MidLFHE.glwe<{_,_,_}{2}>):  // no predecessors
+//CHECK-NEXT:       %4 = "MidLFHE.mul_glwe_int"(%arg2, %arg3) : (!MidLFHE.glwe<{_,_,_}{2}>, i3) -> !MidLFHE.glwe<{_,_,_}{2}>
+//CHECK-NEXT:       %5 = "MidLFHE.add_glwe"(%4, %arg4) : (!MidLFHE.glwe<{_,_,_}{2}>, !MidLFHE.glwe<{_,_,_}{2}>) -> !MidLFHE.glwe<{_,_,_}{2}>
+//CHECK-NEXT:       linalg.yield %5 : !MidLFHE.glwe<{_,_,_}{2}>
+//CHECK-NEXT:     } -> tensor<1x!MidLFHE.glwe<{_,_,_}{2}>>
 //CHECK-NEXT:     %c0 = constant 0 : index
-//CHECK-NEXT:     %3 = tensor.extract %2[%c0] : tensor<1x!HLFHE.eint<2>>
-//CHECK-NEXT:     return %3 : !HLFHE.eint<2>
+//CHECK-NEXT:     %3 = tensor.extract %2[%c0] : tensor<1x!MidLFHE.glwe<{_,_,_}{2}>>
+//CHECK-NEXT:     return %3 : !MidLFHE.glwe<{_,_,_}{2}>
 //CHECK-NEXT:   }
-//CHECK-NEXT:  }
+//CHECK-NEXT: }
 func @dot_eint_int(%arg0: tensor<2x!HLFHE.eint<2>>,
                   %arg1: tensor<2xi3>) -> !HLFHE.eint<2>
 {
--- a/compiler/tests/Dialect/HLFHE/types.mlir
+++ b/compiler/tests/Dialect/HLFHE/types.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --round-trip %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=hlfhe --action=roundtrip %s  2>&1| FileCheck %s

 // CHECK-LABEL: func @memref_arg(%arg0: memref<2x!HLFHE.eint<7>>
 func @memref_arg(%arg0: memref<2x!HLFHE.eint<7>>) {
--- a/compiler/tests/Dialect/LowLFHE/ops.mlir
+++ b/compiler/tests/Dialect/LowLFHE/ops.mlir
@@ -1,5 +1,4 @@
-// RUN: zamacompiler --round-trip %s  2>&1| FileCheck %s
-
+// RUN: zamacompiler --entry-dialect=lowlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: func @add_lwe_ciphertexts(%arg0: !LowLFHE.lwe_ciphertext<2048,7>, %arg1: !LowLFHE.lwe_ciphertext<2048,7>) -> !LowLFHE.lwe_ciphertext<2048,7>
 func @add_lwe_ciphertexts(%arg0: !LowLFHE.lwe_ciphertext<2048,7>, %arg1: !LowLFHE.lwe_ciphertext<2048,7>) -> !LowLFHE.lwe_ciphertext<2048,7> {
--- a/compiler/tests/Dialect/LowLFHE/types.mlir
+++ b/compiler/tests/Dialect/LowLFHE/types.mlir
@@ -1,5 +1,4 @@
-// RUN: zamacompiler --round-trip %s  2>&1| FileCheck %s
-
+// RUN: zamacompiler --entry-dialect=lowlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: func @type_enc_rand_gen(%arg0: !LowLFHE.enc_rand_gen) -> !LowLFHE.enc_rand_gen
 func @type_enc_rand_gen(%arg0: !LowLFHE.enc_rand_gen) -> !LowLFHE.enc_rand_gen {
--- a/compiler/tests/Dialect/MidLFHE/op_add_glwe.invalid.mlir
+++ b/compiler/tests/Dialect/MidLFHE/op_add_glwe.invalid.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --split-input-file --verify-diagnostics %s
+// RUN: zamacompiler --split-input-file --verify-diagnostics --entry-dialect=midlfhe --action=roundtrip %s

 // GLWE p parameter result
 func @add_glwe(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>, %arg1: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{6}> {
--- a/compiler/tests/Dialect/MidLFHE/op_add_glwe.mlir
+++ b/compiler/tests/Dialect/MidLFHE/op_add_glwe.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --round-trip %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=midlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: func @add_glwe(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>, %arg1: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{7}>
 func @add_glwe(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>, %arg1: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{7}> {
--- a/compiler/tests/Dialect/MidLFHE/op_add_glwe_int.invalid.mlir
+++ b/compiler/tests/Dialect/MidLFHE/op_add_glwe_int.invalid.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --split-input-file --verify-diagnostics %s
+// RUN: zamacompiler --split-input-file --verify-diagnostics --entry-dialect=midlfhe --action=roundtrip %s

 // GLWE p parameter
 func @add_glwe_int(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{6}> {
--- a/compiler/tests/Dialect/MidLFHE/op_add_glwe_int.mlir
+++ b/compiler/tests/Dialect/MidLFHE/op_add_glwe_int.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --round-trip %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=midlfhe --action=roundtrip %s  2>&1| FileCheck %s

 // CHECK-LABEL: func @add_glwe_int(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{7}>
 func @add_glwe_int(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{7}> {
@@ -9,4 +9,4 @@ func @add_glwe_int(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024
  %0 = constant 1 : i8
  %1 = "MidLFHE.add_glwe_int"(%arg0, %0): (!MidLFHE.glwe<{1024,12,64}{7}>, i8) -> (!MidLFHE.glwe<{1024,12,64}{7}>)
  return %1: !MidLFHE.glwe<{1024,12,64}{7}>
-}
+}
--- a/compiler/tests/Dialect/MidLFHE/op_apply_lookup_table.invalid.mlir
+++ b/compiler/tests/Dialect/MidLFHE/op_apply_lookup_table.invalid.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --split-input-file --verify-diagnostics %s
+// RUN: zamacompiler --split-input-file --verify-diagnostics --entry-dialect=midlfhe --action=roundtrip %s

 // Bad dimension of the lookup table
 func @apply_lookup_table(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>, %arg1: tensor<4xi2>) -> !MidLFHE.glwe<{512,10,64}{2}> {
--- a/compiler/tests/Dialect/MidLFHE/op_apply_lookup_table.mlir
+++ b/compiler/tests/Dialect/MidLFHE/op_apply_lookup_table.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --round-trip %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=midlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: func @apply_lookup_table(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>, %arg1: tensor<128xi2>) -> !MidLFHE.glwe<{512,10,64}{2}>
 func @apply_lookup_table(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>, %arg1: tensor<128xi2>) -> !MidLFHE.glwe<{512,10,64}{2}> {
@@ -7,4 +7,4 @@ func @apply_lookup_table(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>, %arg1: tensor<12

  %1 = "MidLFHE.apply_lookup_table"(%arg0, %arg1) {k = 1 : i32, polynomialSize = 1024 : i32, levelKS = 2 : i32, baseLogKS = -82 : i32, levelBS = 3 : i32, baseLogBS = -83 : i32, outputSizeKS = 600 : i32} : (!MidLFHE.glwe<{1024,12,64}{7}>, tensor<128xi2>) -> (!MidLFHE.glwe<{512,10,64}{2}>)
  return %1: !MidLFHE.glwe<{512,10,64}{2}>
-}
+}
--- a/compiler/tests/Dialect/MidLFHE/op_mul_glwe_int.invalid.mlir
+++ b/compiler/tests/Dialect/MidLFHE/op_mul_glwe_int.invalid.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --split-input-file --verify-diagnostics %s
+// RUN: zamacompiler --split-input-file --verify-diagnostics --entry-dialect=midlfhe --action=roundtrip %s

 // GLWE p parameter
 func @mul_glwe_int(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{6}> {
--- a/compiler/tests/Dialect/MidLFHE/op_mul_glwe_int.mlir
+++ b/compiler/tests/Dialect/MidLFHE/op_mul_glwe_int.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --round-trip %s  2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=midlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: func @mul_glwe_int(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{7}>
 func @mul_glwe_int(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{7}> {
@@ -9,4 +9,4 @@ func @mul_glwe_int(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024
  %0 = constant 1 : i8
  %1 = "MidLFHE.mul_glwe_int"(%arg0, %0): (!MidLFHE.glwe<{1024,12,64}{7}>, i8) -> (!MidLFHE.glwe<{1024,12,64}{7}>)
  return %1: !MidLFHE.glwe<{1024,12,64}{7}>
-}
+}
--- a/compiler/tests/Dialect/MidLFHE/op_sub_int_glwe.invalid.mlir
+++ b/compiler/tests/Dialect/MidLFHE/op_sub_int_glwe.invalid.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --split-input-file --verify-diagnostics %s
+// RUN: zamacompiler --split-input-file --verify-diagnostics --entry-dialect=midlfhe --action=roundtrip %s

 // GLWE p parameter
 func @sub_int_glwe(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{6}> {
--- a/compiler/tests/Dialect/MidLFHE/op_sub_int_glwe.mlir
+++ b/compiler/tests/Dialect/MidLFHE/op_sub_int_glwe.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler --round-trip %s 2>&1| FileCheck %s
+// RUN: zamacompiler --entry-dialect=midlfhe --action=roundtrip %s 2>&1| FileCheck %s

 // CHECK-LABEL: func @sub_int_glwe(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{7}>
 func @sub_int_glwe(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{7}> {
@@ -9,4 +9,4 @@ func @sub_int_glwe(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024
  %0 = constant 1 : i8
  %1 = "MidLFHE.sub_int_glwe"(%0, %arg0): (i8, !MidLFHE.glwe<{1024,12,64}{7}>) -> (!MidLFHE.glwe<{1024,12,64}{7}>)
  return %1: !MidLFHE.glwe<{1024,12,64}{7}>
-}
+}
--- a/compiler/tests/Dialect/MidLFHE/types_glwe.mlir
+++ b/compiler/tests/Dialect/MidLFHE/types_glwe.mlir
@@ -1,4 +1,4 @@
-// RUN: zamacompiler %s --round-trip 2>&1| FileCheck %s
+// RUN: zamacompiler %s --entry-dialect=midlfhe --action=roundtrip 2>&1| FileCheck %s

 // CHECK-LABEL: func @glwe_0(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{7}>
 func @glwe_0(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64}{7}> {
@@ -10,4 +10,4 @@ func @glwe_0(%arg0: !MidLFHE.glwe<{1024,12,64}{7}>) -> !MidLFHE.glwe<{1024,12,64
 func @glwe_1(%arg0: !MidLFHE.glwe<{_,_,_}{7}>) -> !MidLFHE.glwe<{_,_,_}{7}> {
  // CHECK-LABEL: return %arg0 : !MidLFHE.glwe<{_,_,_}{7}>
  return %arg0: !MidLFHE.glwe<{_,_,_}{7}>
-}
+}