* Ignore terms with 0 coefficients

Signed-off-by: Sachin Pisal <[email protected]>
NVIDIA · Nov 8, 2024 · c20ab81 · c20ab81
1 parent aef4036
commit c20ab81
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/runtime/cudaq/spin/spin_op.cpp b/runtime/cudaq/spin/spin_op.cpp
@@ -414,9 +414,11 @@ spin_op &spin_op::operator+=(const spin_op &v) noexcept {
 
   for (auto [term, coeff] : tmpv.terms) {
     auto iter = terms.find(term);
-    if (iter != terms.end())
+    if (iter != terms.end()) {
       iter->second += coeff;
-    else
+      if (std::abs(iter->second) < 1e-12)
+        terms.erase(iter);
+    } else
       terms.emplace(term, coeff);
   }