fix test

Author: lmondada

test/Transforms/BasisConversion/all-qir-gates.qke

@@ -304,7 +304,8 @@ module {
 // CHECK:           %[[VAL_87:.*]] = cc.cast unsigned %[[VAL_85]] : (i1) -> i8
 // CHECK:           cc.store %[[VAL_87]], %[[VAL_86]] : !cc.ptr<i8>
 // CHECK:           %[[VAL_88:.*]] = quake.alloca !quake.ref
-// CHECK:           quake.rz<adj> (%[[VAL_4]]) %[[VAL_88]] : (f64, !quake.ref) -> ()
+// CHECK:           %[[VAL_89:.*]] = arith.negf %[[VAL_4]] : f64
+// CHECK:           quake.rz (%[[VAL_89]]) %[[VAL_88]] : (f64, !quake.ref) -> ()
 // CHECK:           quake.rx<adj> (%[[VAL_0]]) %[[VAL_88]] : (f64, !quake.ref) -> ()
 // CHECK:           quake.ry<adj> (%[[VAL_3]]) %[[VAL_88]] : (f64, !quake.ref) -> ()
 // CHECK:           quake.rz<adj> (%[[VAL_2]]) %[[VAL_88]] : (f64, !quake.ref) -> ()

unittests/Optimizer/DecompositionPatternsTest.cpp

@@ -44,11 +44,29 @@ class DecompositionPatternsTest : public ::testing::Test {
   std::unique_ptr<MLIRContext> context;
 };
 
-// Helper to parse control count from gate string like "x(1)" or "z(2)"
-std::pair<std::string, size_t> parseGateSpec(StringRef gateSpec) {
+// Helper to parse gate spec like "x(1)", "z(2)", "rx<adj>", "rx<adj>(1)"
+struct GateSpec {
+  std::string name;
+  size_t numControls;
+  bool isAdj;
+};
+
+GateSpec parseGateSpec(StringRef gateSpec) {
+  bool isAdj = false;
+
+  // Strip <adj> if present
+  auto adjPos = gateSpec.find("<adj>");
+  std::string specStr;
+  if (adjPos != StringRef::npos) {
+    isAdj = true;
+    specStr =
+        gateSpec.substr(0, adjPos).str() + gateSpec.substr(adjPos + 5).str();
+    gateSpec = specStr;
+  }
+
   auto pos = gateSpec.find('(');
   if (pos == StringRef::npos) {
-    return {gateSpec.str(), 0};
+    return {gateSpec.str(), 0, isAdj};
   }
 
   std::string gateName = gateSpec.substr(0, pos).str();
@@ -62,12 +80,12 @@ std::pair<std::string, size_t> parseGateSpec(StringRef gateSpec) {
     numStr.consumeInteger(10, numControls);
   }
 
-  return {gateName, numControls};
+  return {gateName, numControls, isAdj};
 }
 
 // Helper function to create a test module with a single gate operation
 ModuleOp createTestModule(MLIRContext *context, StringRef gateSpec) {
-  auto [gateName, numControls] = parseGateSpec(gateSpec);
+  auto [gateName, numControls, isAdj] = parseGateSpec(gateSpec);
 
   // Limit the number of controls to 2
   numControls = std::min<size_t>(numControls, 2);
@@ -161,6 +179,13 @@ ModuleOp createTestModule(MLIRContext *context, StringRef gateSpec) {
     ADD_FAILURE() << "unknown gate: " << gateName;
   }
 
+  // Set adjoint attribute on the last-created quantum op if needed
+  if (isAdj) {
+    auto &lastOp = entry->back();
+    if (isa<quake::OperatorInterface>(&lastOp))
+      lastOp.setAttr("is_adj", builder.getUnitAttr());
+  }
+
   builder.create<func::ReturnOp>(loc);
   return module;
 }