Merge pull request #66 from sickkids-mri/memleak

Fix memory leak caused by alg referencing app object
mikgroup · Oct 27, 2020 · 8c25c53 · 8c25c53
2 parents ac449e7 + 11dc18f
commit 8c25c53
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sigpy/mri/app.py b/sigpy/mri/app.py
@@ -448,11 +448,11 @@ def __init__(self, ksp, calib_width=24,
             self.mps = xp.ones(ksp.shape[::-1] + (1, ), dtype=ksp.dtype)
 
             def forward(x):
-                with self.device:
+                with sp.get_device(x):
                     return AHA @ x
 
             def normalize(x):
-                with self.device:
+                with sp.get_device(x):
                     return xp.sum(xp.abs(x)**2,
                                   axis=-2, keepdims=True)**0.5