Update discord.hpp

Partial trace in disc_nlopt3 fixed.
titaschanda · Oct 17, 2023 · 87d61a7 · 87d61a7
1 parent 47e565c
commit 87d61a7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/include/QIClib_bits/internal/discord.hpp b/include/QIClib_bits/internal/discord.hpp
@@ -197,15 +197,15 @@ inline double disc_nlopt3(const std::vector<double>& x,
   trait::pT<T1> S_max = 0.0;
   if (p1 > _precision::eps<trait::pT<T1> >::value) {
     rho_1 /= p1;
-    S_max += p1 * entropy(TrX(rho_1, {pB->nodal}, pB->nodal));
+    S_max += p1 * entropy(TrX(rho_1, {pB->nodal}, pB->dim));
   }
   if (p2 > _precision::eps<trait::pT<T1> >::value) {
     rho_2 /= p2;
-    S_max += p2 * entropy(TrX(rho_2, {pB->nodal}, pB->nodal));
+    S_max += p2 * entropy(TrX(rho_2, {pB->nodal}, pB->dim));
   }
   if (p3 > _precision::eps<trait::pT<T1> >::value) {
     rho_3 /= p3;
-    S_max += p3 * entropy(TrX(rho_3, {pB->nodal}, pB->nodal));
+    S_max += p3 * entropy(TrX(rho_3, {pB->nodal}, pB->dim));
   }
 
   return static_cast<double>(S_max);