add hasCompileTimeKnownUnitaryMatrix

Author: J4MMlE

mlir/include/mlir/Dialect/QCO/IR/QCOOps.td

@@ -1274,6 +1274,7 @@ def PowOp
         OperandRange getParameters() { return getBodyUnitary().getParameters(); }
         double getExponentValue();
         [[nodiscard]] static StringRef getBaseSymbol() { return "pow"; }
+        [[nodiscard]] bool hasCompileTimeKnownUnitaryMatrix();
         [[nodiscard]] std::optional<DynamicMatrix> getUnitaryMatrix();
     }];
 

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

@@ -689,6 +689,13 @@ void PowOp::getCanonicalizationPatterns(RewritePatternSet& results,
               MoveCtrlOutside, NegPowToInvPow>(context);
 }
 
+bool PowOp::hasCompileTimeKnownUnitaryMatrix() {
+  return all_of(getBody()->getOps<UnitaryOpInterface>(),
+                [](UnitaryOpInterface op) {
+                  return op.hasCompileTimeKnownUnitaryMatrix();
+                });
+}
+
 std::optional<DynamicMatrix> PowOp::getUnitaryMatrix() {
   auto bodyUnitary = utils::getSoleBodyUnitary<UnitaryOpInterface>(*getBody());
   if (!bodyUnitary) {