Fixed VQE missing basis rotation caching in execute(buffer, vector)

Signed-off-by: Daniel Claudino <[email protected]>
ORNL-QCI · Dec 10, 2024 · 689ba43 · 689ba43
1 parent 7d4fb9f
commit 689ba43
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);