Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[BUG] Dask PCA #5883

Open
Intron7 opened this issue May 3, 2024 · 2 comments
Open

[BUG] Dask PCA #5883

Intron7 opened this issue May 3, 2024 · 2 comments
Labels
? - Needs Triage Need team to review and classify bug Something isn't working

Comments

@Intron7
Copy link
Contributor

Intron7 commented May 3, 2024

Describe the bug
The return value element of transform has no shape and cant be integrated into a existing data structure without calling PCA.compute_chuch_sizes() because its has no size.

Steps/Code to reproduce bug

from dask_cuda import LocalCUDACluster
from dask.distributed import Client, wait
import cupy as cp
from cuml.dask.decomposition import PCA
from cuml.dask.datasets import make_blobs

cluster = LocalCUDACluster(threads_per_worker=1)
client = Client(cluster)

nrows = 6
ncols = 3
n_parts = 2

X_cudf, _ = make_blobs(n_samples=nrows, n_features=ncols,
                       centers=1, n_parts=n_parts,
                       cluster_std=0.01, random_state=10,
                       dtype=cp.float32)


cumlModel = PCA(n_components = 1, whiten=False)
XT = cumlModel.fit_transform(X_cudf)
print(XT.shape)

Expected behavior
A clear and concise description of what you expected to happen.

Environment details (please complete the following information):

  • Environment location: Bare-metal
  • Linux Distro/Architecture: Ubuntu 22.04 amd64
  • GPU Model/Driver: [3090 and driver 550.54.15]
  • CUDA: [11.8]
@Intron7 Intron7 added ? - Needs Triage Need team to review and classify bug Something isn't working labels May 3, 2024
@dantegd
Copy link
Member

dantegd commented May 3, 2024

@Intron7 thanks for reporting this! I'll try to repro and find a fix as soon as we can.

@Intron7
Copy link
Contributor Author

Intron7 commented May 3, 2024

@dantegd I have an easy workaround for this #5555 is way more important imo.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
? - Needs Triage Need team to review and classify bug Something isn't working
Projects
None yet
Development

No branches or pull requests

2 participants