Add ExtractCircuitsPass

include/Dialect/QUIR/Transforms/ExtractCircuits.h

@@ -0,0 +1,70 @@
+//===- ExtractCircuits.h - Extract circuits ops -----------------*- C++ -*-===//
+//
+// (C) Copyright IBM 2024.
+//
+// This code is part of Qiskit.
+//
+// This code is licensed under the Apache License, Version 2.0 with LLVM
+// Exceptions. You may obtain a copy of this license in the LICENSE.txt
+// file in the root directory of this source tree.
+//
+// Any modifications or derivative works of this code must retain this
+// copyright notice, and modified files need to carry a notice indicating
+// that they have been altered from the originals.
+//
+//===----------------------------------------------------------------------===//
+///
+///  This file declares the pass for extracting quantum ops into quir.circuits
+///
+//===----------------------------------------------------------------------===//
+
+#ifndef QUIR_EXTRACT_CIRCUITS_H
+#define QUIR_EXTRACT_CIRCUITS_H
+
+#include "Dialect/QUIR/IR/QUIROps.h"
+
+#include "mlir/IR/PatternMatch.h"
+#include "mlir/Pass/Pass.h"
+
+#include "llvm/ADT/SmallVector.h"
+
+#include <unordered_map>
+
+namespace mlir::quir {
+
+struct ExtractCircuitsPass
+    : public PassWrapper<ExtractCircuitsPass, OperationPass<>> {
+  void runOnOperation() override;
+
+  llvm::StringRef getArgument() const override;
+  llvm::StringRef getDescription() const override;
+  llvm::StringRef getName() const override;
+
+private:
+  void processOps(mlir::Operation *currentOp, OpBuilder topLevelBuilder,
+                  OpBuilder circuitBuilder);
+  OpBuilder startCircuit(mlir::Location location, OpBuilder topLevelBuilder);
+  void endCircuit(mlir::Operation *firstOp, mlir::Operation *lastOp,
+                  OpBuilder topLevelBuilder, OpBuilder circuitBuilder,
+                  llvm::SmallVector<Operation *> &eraseList);
+  void addToCircuit(mlir::Operation *currentOp, OpBuilder circuitBuilder,
+                    llvm::SmallVector<Operation *> &eraseList);
+  uint64_t circuitCount;
+  llvm::StringMap<Operation *> circuitOpsMap;
+
+  mlir::quir::CircuitOp currentCircuitOp;
+  mlir::quir::CallCircuitOp newCallCircuitOp;
+
+  llvm::SmallVector<Type> inputTypes;
+  llvm::SmallVector<Value> inputValues;
+  llvm::SmallVector<Type> outputTypes;
+  llvm::SmallVector<Value> outputValues;
+  std::vector<int> phyiscalIds;
+
+  std::unordered_map<uint32_t, BlockArgument> circuitArguments;
+  std::unordered_map<Operation *, uint32_t> circuitOperands;
+  llvm::SmallVector<OpResult> originalResults;
+
+}; // struct ExtractCircuitsPass
+} // namespace mlir::quir
+#endif // QUIR_EXTRACT_CIRCUITS_H

include/Dialect/QUIR/Transforms/Passes.h

@@ -21,6 +21,7 @@
 #include "AngleConversion.h"
 #include "BreakReset.h"
 #include "ConvertDurationUnits.h"
+#include "ExtractCircuits.h"
 #include "FunctionArgumentSpecialization.h"
 #include "LoadElimination.h"
 #include "MergeCircuitMeasures.h"

lib/Dialect/QUIR/Transforms/CMakeLists.txt

@@ -16,6 +16,7 @@ add_mlir_dialect_library(MLIRQUIRTransforms
     AngleConversion.cpp
     BreakReset.cpp
     ConvertDurationUnits.cpp
+    ExtractCircuits.cpp
     FunctionArgumentSpecialization.cpp
     LoadElimination.cpp
     MergeCircuits.cpp