Lower ion to rtio

doc/releases/changelog-dev.md

@@ -2,8 +2,9 @@
 
 <h3>New features since last release</h3>
 
-* RTIO dialect is added to bypass the compilation flow from OpenAPL to ARTIQ’s LLVM IR. It is introduced to bridge the gap between ION dialect and ARTIQ’s LLVM IR. The design philosophy of RTIO dialect is primarily event-based. Every operation is asynchronous; sync behaviour occurs only via `rtio.sync` or `wait operand` in event operation.
+* RTIO dialect is added to bypass the compilation flow from OpenAPL to ARTIQ’s LLVM IR. It is introduced to bridge the gap between ION dialect and ARTIQ’s LLVM IR. The design philosophy of RTIO dialect is primarily event-based. Every operation is asynchronous; sync behaviour occurs only via `rtio.sync` or `wait operand` in event operation. And we now support the compiling from ION dialect to RTIO dilalect.
   [(#2185)](https://github.com/PennyLaneAI/catalyst/pull/2185)
+  [(#2204)](https://github.com/PennyLaneAI/catalyst/pull/2204)
 
 * Added ``catalyst.switch``, a qjit compatible, index-switch style control flow decorator.
   [(#2171)](https://github.com/PennyLaneAI/catalyst/pull/2171)

mlir/include/Ion/Transforms/Passes.td

@@ -52,4 +52,28 @@ def IonConversionPass : Pass<"convert-ion-to-llvm"> {
     ];
 }
 
+def IonToRTIOPass : Pass<"convert-ion-to-rtio", "mlir::ModuleOp"> {
+    let summary = "Convert Ion dialect operations to RTIO dialect";
+
+    let dependentDialects = [
+        "rtio::RTIODialect",
+        "arith::ArithDialect",
+        "func::FuncDialect",
+        "memref::MemRefDialect",
+        "scf::SCFDialect",
+        "quantum::QuantumDialect",
+        "ion::IonDialect",
+        "linalg::LinalgDialect",
+    ];
+
+    let options = [
+        Option<"kernelName", "kernel-name",
+               "std::string", /*default=*/"\"__kernel__\"",
+               "Name of the generated kernel function">,
+        Option<"deviceDb", "device_db",
+               "std::string", /*default=*/"\"\"",
+               "Path to the device database JSON file">,
+    ];
+}
+
 #endif // ION_PASSES

mlir/include/RTIO/IR/RTIOOps.h

@@ -18,6 +18,7 @@
 #include "mlir/Bytecode/BytecodeOpInterface.h"
 #include "mlir/IR/BuiltinTypes.h"
 #include "mlir/IR/OpDefinition.h"
+#include "mlir/IR/PatternMatch.h"
 #include "mlir/Interfaces/SideEffectInterfaces.h"
 
 #define GET_ATTRDEF_CLASSES

mlir/include/RTIO/IR/RTIOOps.td

@@ -20,7 +20,7 @@ include "mlir/IR/BuiltinAttributes.td"
 include "mlir/Interfaces/SideEffectInterfaces.td"
 include "RTIO/IR/RTIODialect.td"
 
-def RTIOChannelOp : RTIO_Op<"channel"> {
+def RTIOChannelOp : RTIO_Op<"channel", [Pure]> {
     let summary = "Define a channel";
     let description = [{
         The channel's identity (kind, qualifiers, and channel id) is
@@ -51,7 +51,7 @@ def RTIOChannelOp : RTIO_Op<"channel"> {
     }];
 }
 
-def RTIOQubitToChannelOp : RTIO_Op<"qubit_to_channel"> {
+def RTIOQubitToChannelOp : RTIO_Op<"qubit_to_channel", [Pure]> {
     let summary = "Map a qubit to an RTIO channel";
     let description = [{
         It's a temporary operation that will be lowered to a static `rtio.channel` operation during

mlir/lib/Ion/Transforms/CMakeLists.txt

@@ -2,6 +2,7 @@ set(LIBRARY_NAME ion-transforms)
 
 file(GLOB SRC
     ion-to-llvm.cpp
+    ion-to-rtio.cpp
     ConversionPatterns.cpp
     gates_to_pulses.cpp
     GatesToPulsesPatterns.cpp
@@ -13,6 +14,7 @@ set(LIBS
     ${dialect_libs}
     ${conversion_libs}
     MLIRIon
+    MLIRRTIO
 )
 
 set(DEPENDS
@@ -36,6 +38,7 @@ target_link_libraries(${LIBRARY_NAME} PRIVATE
 if(CMAKE_CXX_COMPILER_ID MATCHES ".*Clang")
 set_source_files_properties(
     ion-to-llvm.cpp
+    ion-to-rtio.cpp
     ConversionPatterns.cpp
     gates_to_pulses.cpp
     GatesToPulsesPatterns.cpp