Plug memory leak in CG. (patch by João Muranho)

inducer · Nov 23, 2009 · e4c9833 · e4c9833
1 parent abc23a5
commit e4c9833
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/cg.hpp b/src/cg.hpp
@@ -98,7 +98,8 @@ namespace iterative_cuda
       a(q, d);
 
       gpu_scalar_t alpha(1);
-      divide(alpha, *delta_new, *d.dot(q));
+      std::auto_ptr<gpu_scalar_t> d_dot_q(d.dot(q));
+      divide(alpha, *delta_new, *d_dot_q);
 
       x.set_to_linear_combination(1, x, 1, alpha, d);