Fixed max_eig device placement in LinearLeastSquares.

mikgroup · Aug 22, 2018 · 0ed2b48 · 0ed2b48
1 parent aeb5c8f
commit 0ed2b48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sigpy/app.py b/sigpy/app.py
@@ -175,12 +175,12 @@ def _init(self):
 
         elif isinstance(self.alg, GradientMethod):
             try:
-                self.alg.alpha = 1 / self.max_eig_app.run()
+                self.alg.alpha = 1 / util.move(self.max_eig_app.run())
             except AttributeError:
                 pass
         elif isinstance(self.alg, PrimalDualHybridGradient):
             try:
-                self.alg.sigma = 1 / self.max_eig_app.run()
+                self.alg.sigma = 1 / util.move(self.max_eig_app.run())
                 self.alg.tau = 1
             except AttributeError:
                 pass