Bugs/1258 bug lasso does not work on gpu (#1269)

* corrected bug * Fix Pytorch release tracking workflows (#1264) * upgrade checkout action & use default token * increase tolerance for single-prec torch.inv comparison * fix typo --------- Co-authored-by: Claudia Comito <[email protected]> --------- Co-authored-by: Hoppe <[email protected]> Co-authored-by: Michael Tarnawa <[email protected]> Co-authored-by: Claudia Comito <[email protected]>
helmholtz-analytics · Nov 22, 2023 · 5125ccc · 5125ccc
1 parent 29213dc
commit 5125ccc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/heat/regression/lasso.py b/heat/regression/lasso.py
@@ -149,7 +149,7 @@ def fit(self, x: DNDarray, y: DNDarray) -> None:
 
             # Looping through each coordinate
             for j in range(n):
-                X_j = ht.array(x.larray[:, j : j + 1], is_split=0)
+                X_j = ht.array(x.larray[:, j : j + 1], is_split=0, device=x.device, comm=x.comm)
 
                 y_est = x @ theta
                 theta_j = theta.larray[j].item()