Fix handling cast without parameter (#328)

openqasm · May 20, 2024 · 7ac5f99 · 7ac5f99
1 parent 166b097
commit 7ac5f99
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/Conversion/QUIRToPulse/QUIRToPulse.cpp b/lib/Conversion/QUIRToPulse/QUIRToPulse.cpp
@@ -542,6 +542,12 @@ mlir::Value QUIRToPulsePass::convertAngleToF64(Operation *angleOp,
             paramCastOp->getLoc(), builder.getF64Type(), paramCastOp.getRes());
         angleCastedOp->moveAfter(paramCastOp);
         classicalQUIROpLocToConvertedPulseOpMap[angleLocHash] = angleCastedOp;
+      } else if (auto constOp =
+                     dyn_cast<arith::ConstantOp>(castOpArg.getDefiningOp())) {
+        // if cast from float64 then use directly
+        assert(constOp.getType() == builder.getF64Type() &&
+               "expected angle type to be float 64");
+        classicalQUIROpLocToConvertedPulseOpMap[angleLocHash] = constOp;
       } else
         llvm_unreachable("castOp arg unknown");
     } else