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

Use 64 bit CuSolver API for Eigen decomposition #349

Merged
merged 10 commits into from
Nov 5, 2021

Conversation

lowener
Copy link
Contributor

@lowener lowener commented Oct 1, 2021

This PR replace the legacy call to cusolverDnsyevj with the 64-bit version.
It also improves the indexing types used, from int to size_t when possible.

@lowener lowener requested review from a team as code owners October 1, 2021 14:48
@github-actions github-actions bot added the cpp label Oct 1, 2021
@lowener
Copy link
Contributor Author

lowener commented Oct 1, 2021

I just noticed that DnXsyevd was introduced in CUDA 11.1.0 in Sept. 2020. So the CI with CUDA version 11.0 will always fail.

@cjnolet
Copy link
Member

cjnolet commented Oct 4, 2021

@lowener,

I just noticed that DnXsyevd was introduced in CUDA 11.1.0 in Sept. 2020. So the CI with CUDA version 11.0 will always fail.

Indeed, I remember discussing this very detail w/ @teju85 not too long ago. A good workaround here could be to be check the CUDA version and have the wrapper call the old version in 11.1.

@lowener lowener added breaking Breaking change improvement Improvement / enhancement to an existing function labels Oct 5, 2021
cpp/include/raft/linalg/eig.cuh Outdated Show resolved Hide resolved
@lowener lowener requested a review from cjnolet October 22, 2021 12:06
@cjnolet
Copy link
Member

cjnolet commented Oct 22, 2021

rerun tests

@cjnolet
Copy link
Member

cjnolet commented Oct 26, 2021

@lowener, I'm ready to merge this over but I'm waiting to see the CI pass on rapidsai/cuml#4255 first.

@cjnolet
Copy link
Member

cjnolet commented Nov 5, 2021

@gpucibot merge

@rapids-bot rapids-bot bot merged commit 3848fa7 into rapidsai:branch-21.12 Nov 5, 2021
@lowener lowener deleted the 21.12-eig-64 branch November 5, 2021 18:14
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
breaking Breaking change cpp improvement Improvement / enhancement to an existing function
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants