Merge pull request #9 from dantegd/fix-unique-array

[HOTFIX] Unique parameter to avoid hash collision wiht higher number of GPUs
dantegd · Mar 17, 2019 · 6246aaa · 6246aaa
2 parents 75d2196 + 22d0a56
commit 6246aaa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dask_cuml/linear_model/linear_regression.py b/dask_cuml/linear_model/linear_regression.py
@@ -112,6 +112,7 @@ def _do_fit(self, X_df, y_df, dtype):
             coefs.append(client.submit(dev_array_on_worker,
                                        up_limit - i*part_size,
                                        dtype=dtype,
+                                       unique=np.random.randint(0, 1e6),
                                        workers=[loc_dict[i]]))
             yield wait(coefs)
             del(loc_cudf)