Merge pull request #59 from danclaudino/vqe_basis_rot_bugfix

Fixed VQE missing basis rotation caching in execute(buffer, vector)
ORNL-QCI · Dec 11, 2024 · 8e4cba7 · 8e4cba7
2 parents 83e5e16 + 689ba43
commit 8e4cba7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/quantum/plugins/algorithms/vqe/vqe.cpp b/quantum/plugins/algorithms/vqe/vqe.cpp
@@ -342,6 +342,10 @@ VQE::execute(const std::shared_ptr<AcceleratorBuffer> buffer,
   // circuits
   if (cacheMeasurements) {
 
+    if (basisRotations.empty()) {
+      basisRotations = observable->getMeasurementBasisRotations();
+    }
+
     nInstructionsEnergy = basisRotations.size() - 1;
     for (auto it = basisRotations.begin(); it != basisRotations.end();) {
 
@@ -387,6 +391,7 @@ VQE::execute(const std::shared_ptr<AcceleratorBuffer> buffer,
   } else {
     accelerator->execute(tmpBuffer, fsToExec);
   }
+
   auto buffers = tmpBuffer->getChildren();
   for (auto &b : buffers) {
     b->addExtraInfo("parameters", x);