add orion-to-ckks conversion

Author: j2kun

MODULE.bazel.lock

@@ -27,9 +27,7 @@ def Mgmt_ModReduceOp : Mgmt_Op<"modreduce"> {
     or ckks.rescale depending on the scheme.
   }];
 
-  let arguments = (ins
-    AnyType:$input
-  );
+  let arguments = (ins AnyType:$input);
   let results = (outs AnyType:$output);
   let assemblyFormat = "operands attr-dict `:` type($output)";
 
@@ -81,9 +79,7 @@ def Mgmt_RelinearizeOp : Mgmt_Op<"relinearize"> {
     or ckks.relinearize depending on the scheme.
   }];
 
-  let arguments = (ins
-    AnyType:$input
-  );
+  let arguments = (ins AnyType:$input);
   let results = (outs AnyType:$output);
   let assemblyFormat = "operands attr-dict `:` type($output)";
 }
@@ -99,16 +95,12 @@ def Mgmt_BootstrapOp : Mgmt_Op<"bootstrap"> {
     reduce the noise in a ciphertext, allowing further operations
     to be performed on it without decryption.
 
-    When further lowered, it could be lowered to bgv.bootstrap
-    or ckks.bootstrap depending on the scheme.
-
-    For the current backend, only ckks.bootstrap is supported.
-    Further backend may include bgv.bootstrap.
+    When further lowered, it could be lowered to `bgv.bootstrap` or
+    `ckks.bootstrap` depending on the scheme. Current backends only support
+    `ckks.bootstrap`.
   }];
 
-  let arguments = (ins
-    AnyType:$input
-  );
+  let arguments = (ins AnyType:$input);
   let results = (outs AnyType:$output);
   let assemblyFormat = "operands attr-dict `:` type($output)";
 }
@@ -175,9 +167,7 @@ def Mgmt_InitOp : Mgmt_Op<"init",
     ConditionallySpeculatable is for isSpeculatable check in hoisting canonicalization.
   }];
 
-  let arguments = (ins
-    AnyType:$input
-  );
+  let arguments = (ins AnyType:$input);
   let results = (outs AnyType:$output);
   let assemblyFormat = "operands attr-dict `:` type($output)";
 

lib/Dialect/Mgmt/IR/MgmtOps.td

@@ -0,0 +1,54 @@
+load("@heir//lib/Transforms:transforms.bzl", "add_heir_transforms")
+
+package(
+    default_applicable_licenses = ["@heir//:license"],
+    default_visibility = ["//visibility:public"],
+)
+
+cc_library(
+    name = "OrionToCKKS",
+    srcs = ["OrionToCKKS.cpp"],
+    hdrs = ["OrionToCKKS.h"],
+    deps = [
+        ":IRMaterializingVisitor",
+        ":pass_inc_gen",
+        "@heir//lib/Dialect/CKKS/IR:Dialect",
+        "@heir//lib/Dialect/LWE/IR:Dialect",
+        "@heir//lib/Dialect/Orion/IR:Dialect",
+        "@heir//lib/Kernel:ArithmeticDag",
+        "@heir//lib/Kernel:IRMaterializingVisitor",
+        "@heir//lib/Kernel:KernelImplementation",
+        "@heir//lib/Utils:ConversionUtils",
+        "@heir//lib/Utils/Polynomial:ChebyshevPatersonStockmeyer",
+        "@llvm-project//mlir:IR",
+        "@llvm-project//mlir:Pass",
+        "@llvm-project//mlir:Transforms",
+    ],
+)
+
+cc_library(
+    name = "IRMaterializingVisitor",
+    srcs = ["IRMaterializingVisitor.cpp"],
+    hdrs = ["IRMaterializingVisitor.h"],
+    deps = [
+        "@heir//lib/Dialect/CKKS/IR:Dialect",
+        "@heir//lib/Dialect/LWE/IR:Dialect",
+        "@heir//lib/Dialect/TensorExt/IR:Dialect",
+        "@heir//lib/Kernel:AbstractValue",
+        "@heir//lib/Kernel:ArithmeticDag",
+        "@heir//lib/Kernel:KernelImplementation",
+        "@heir//lib/Utils",
+        "@heir//lib/Utils:MathUtils",
+        "@llvm-project//llvm:Support",
+        "@llvm-project//mlir:ArithDialect",
+        "@llvm-project//mlir:IR",
+        "@llvm-project//mlir:Support",
+        "@llvm-project//mlir:TensorDialect",
+    ],
+)
+
+add_heir_transforms(
+    header_filename = "OrionToCKKS.h.inc",
+    pass_name = "OrionToCKKS",
+    td_file = "OrionToCKKS.td",
+)