temporary fix, to be fixed in #1474

Author: taminob

mlir/lib/Dialect/QCO/IR/Modifiers/CtrlOp.cpp

@@ -145,7 +145,9 @@ struct CtrlInlineId final : OpRewritePattern<CtrlOp> {
 } // namespace
 
 UnitaryOpInterface CtrlOp::getBodyUnitary() {
-  return llvm::dyn_cast<UnitaryOpInterface>(&getBody()->front());
+  auto bodyUnitary = llvm::dyn_cast<UnitaryOpInterface>(&getBody()->front());
+  assert(bodyUnitary);
+  return bodyUnitary;
 }
 
 size_t CtrlOp::getNumQubits() { return getNumTargets() + getNumControls(); }
@@ -160,7 +162,7 @@ Value CtrlOp::getInputQubit(const size_t i) {
     return getControlsIn()[i];
   }
   if (numControls <= i && i < getNumQubits()) {
-    return getBodyUnitary().getInputQubit(i - numControls);
+    return getTargetsIn()[i - numControls];
   }
   llvm::reportFatalUsageError("Invalid qubit index");
 }
@@ -171,7 +173,7 @@ Value CtrlOp::getOutputQubit(const size_t i) {
     return getControlsOut()[i];
   }
   if (numControls <= i && i < getNumQubits()) {
-    return getBodyUnitary().getOutputQubit(i - numControls);
+    return getTargetsOut()[i - numControls];
   }
   llvm::reportFatalUsageError("Invalid qubit index");
 }