1 files changed, 5 insertions, 3 deletions
diff --git a/hadamard_pt.hpp b/hadamard_pt.hpp
index dad635f..bddd8b7 100644
--- a/hadamard_pt.hpp
+++ b/hadamard_pt.hpp
@@ -69,14 +69,16 @@ public:
     }
   }
 
-  void run(unsigned n, unsigned m) {
+  void run(unsigned n, unsigned m, bool dry = false) {
     for (unsigned i = 0; i < n; i++) {
 #pragma omp parallel for
       for (unsigned j = 0; j < Ms.size(); j++) {
-        Ms[j].run(m);
+        Ms[j].run(m, dry);
       }
       this->sweep();
-      B.after_sweep(this->Ms);
+      if (!dry) {
+        B.after_sweep(this->Ms);
+      }
     }
   }
 };