dp: fix cost_cap check

jcmgray · Nov 3, 2020 · c75b8e7 · c75b8e7
1 parent ff0fc52
commit c75b8e7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/opt_einsum/paths.py b/opt_einsum/paths.py
@@ -1035,7 +1035,7 @@ def __call__(self, inputs, output, size_dict, memory_limit=None):
                                                                 xn, g, all_tensors, inputs, i1_cut_i2_wo_output,
                                                                 memory_limit, cntrct1, cntrct2)
 
-                if (cost_cap >= naive_cost) and (len(x[-1]) == 0):
+                if (cost_cap > naive_cost) and (len(x[-1]) == 0):
                     raise RuntimeError("No contraction found for given `memory_limit`.")
 
                 # increase cost cap for next iteration: