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

Accelerate LU when input has one chunk #905

Merged
merged 1 commit into from Jan 9, 2020

Conversation

qinxuye
Copy link
Collaborator

@qinxuye qinxuye commented Jan 8, 2020

What do these changes do?

As described in #904 , when input has one chunk, execution of LU is super slow. This PR accelerated the process.

Because LU is used in randomized_svd, and the latter one is widely used in PCA and TruncatedSVD, I test the unittests in test_pca, the running time reduced from 5min 14s to 1min 35s on my laptop. I think this PR can accelerate the CI as well.

Related issue number

Fixes #904 .

@qinxuye qinxuye added this to the v0.4.0a1 milestone Jan 8, 2020
@qinxuye qinxuye added this to In progress in Tensor via automation Jan 8, 2020
@qinxuye qinxuye changed the title Accelerate LU tile when input has one chunk Accelerate LU when input has one chunk Jan 8, 2020
@qinxuye qinxuye added the to be backported Indicate that the PR need to be backported to stable branch label Jan 8, 2020
Tensor automation moved this from In progress to Reviewer approved Jan 9, 2020
Copy link
Member

@wjsi wjsi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@wjsi wjsi merged commit c68ed39 into mars-project:master Jan 9, 2020
Tensor automation moved this from Reviewer approved to Done Jan 9, 2020
wjsi pushed a commit to wjsi/mars that referenced this pull request Jan 9, 2020
wjsi pushed a commit to wjsi/mars that referenced this pull request Jan 9, 2020
@wjsi wjsi added backported already PR has been backported and removed to be backported Indicate that the PR need to be backported to stable branch labels Jan 9, 2020
@qinxuye qinxuye deleted the enh/lu-tile-one-chunk branch January 9, 2020 13:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
Tensor
  
Done
Development

Successfully merging this pull request may close these issues.

LU decomposition is super slow when input has one chunk
2 participants