refactor(compiler): Re-implement TFHE multi-parameter parametrization…

… with type inference

The current pass applying the parameters determined by the optimizer
to the IR propagates the parametrized TFHE types to operations not
directly tagged with an optimizer ID only under certain conditions. In
particular, it does not always properly propagate types into nested
regions (e.g., of `scf.for` loops).

This burdens preceding transformations that are applied in between the
invocation of the optimizer and the parametrization pass with
data-flow analysis and book-keeping in order to tag newly inserted
operations with the right optimizer IDs that ensure proper
parametrization.

This commit replaces the current parametrization pass with a new pass
that propagates parametrized TFHE types up and down def-use chains
using type inference and a proper rewriter. The pass is limited to the
operations supported by `TFHEParametrizationTypeResolver::resolve`.

compilers/concrete-compiler/compiler/include/concretelang/Dialect/TFHE/Transforms/Transforms.h

@@ -16,9 +16,9 @@
 namespace mlir {
 namespace concretelang {
 std::unique_ptr<mlir::OperationPass<>> createTFHEOptimizationPass();
-std::unique_ptr<mlir::OperationPass<>>
+std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>>
     createTFHECircuitSolutionParametrizationPass(
-        concrete_optimizer::dag::CircuitSolution);
+        std::optional<concrete_optimizer::dag::CircuitSolution>);
 } // namespace concretelang
 } // namespace mlir
 

compilers/concrete-compiler/compiler/include/concretelang/Dialect/TFHE/Transforms/Transforms.td

@@ -10,7 +10,7 @@ def TFHEOptimization : Pass<"tfhe-optimization"> {
   let dependentDialects = [ "mlir::concretelang::TFHE::TFHEDialect" ];
 }
 
-def TFHECircuitSolutionParametrization : Pass<"tfhe-circuit-solution-parametrization"> {
+def TFHECircuitSolutionParametrization : Pass<"tfhe-circuit-solution-parametrization", "mlir::ModuleOp"> {
   let summary = "Parametrize TFHE with a circuit solution given by the optimizer";
   let constructor = "mlir::concretelang::createTFHECircuitSolutionParametrizationPass()";
   let options = [];