Skip to content

Commit 66ab8f5

Browse files
fixed bug with GPU in gaussian_approx
1 parent be8face commit 66ab8f5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

PLASIM/gaussian_approx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -213,7 +213,7 @@ def _fit(self, X, A):
213213

214214
if self.GPU:
215215
# convert back to CPU
216-
self.p = np.copy(self.p)
216+
self.p = self.engine.asnumpy(self.p)
217217

218218
# compute the projected coordinate and the rescaling
219219
self.f_tr = X @ self.p

0 commit comments

Comments
 (0)