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

[Lang] Experimental sparse matrix support on CPUs #2792

Merged
merged 77 commits into from
Aug 28, 2021

Conversation

FantasyVR
Copy link
Collaborator

@FantasyVR FantasyVR commented Aug 24, 2021

Related pr = #2657
Related issue = #2906

  • Support sparse matrix builder and sparse matrix class
  • Support some basic sparse matrix operations.

Copy link
Member

@yuanming-hu yuanming-hu left a comment

Choose a reason for hiding this comment

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

Thanks! This is getting closer and closer to merging!

python/taichi/lang/sparse_matrix.py Outdated Show resolved Hide resolved
python/taichi/lang/sparse_matrix.py Outdated Show resolved Hide resolved
python/taichi/lang/sparse_matrix.py Outdated Show resolved Hide resolved
tests/python/test_sparse_matrix.py Outdated Show resolved Hide resolved
tests/python/test_sparse_matrix.py Outdated Show resolved Hide resolved
tests/python/test_sparse_matrix.py Outdated Show resolved Hide resolved
taichi/python/export_lang.cpp Outdated Show resolved Hide resolved
taichi/program/sparse_matrix.cpp Outdated Show resolved Hide resolved
taichi/program/sparse_matrix.cpp Outdated Show resolved Hide resolved
taichi/program/sparse_matrix.h Outdated Show resolved Hide resolved
FantasyVR and others added 3 commits August 26, 2021 22:25
Co-authored-by: Yuanming Hu <yuanming-hu@users.noreply.github.com>
@FantasyVR
Copy link
Collaborator Author

/format

cmake/TaichiCore.cmake Outdated Show resolved Hide resolved
misc/sparse_matrix.py Outdated Show resolved Hide resolved
python/taichi/lang/kernel_impl.py Outdated Show resolved Hide resolved
taichi/python/export_lang.cpp Outdated Show resolved Hide resolved
taichi/python/export_lang.cpp Show resolved Hide resolved
taichi/ir/frontend_ir.h Outdated Show resolved Hide resolved
python/taichi/lang/sparse_matrix.py Outdated Show resolved Hide resolved
python/taichi/lang/sparse_matrix.py Outdated Show resolved Hide resolved
Copy link
Member

@yuanming-hu yuanming-hu left a comment

Choose a reason for hiding this comment

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

My final pass with a few nits. This PR LGTM after these issues are fixed.

misc/sparse_matrix.py Outdated Show resolved Hide resolved
misc/sparse_matrix.py Outdated Show resolved Hide resolved
misc/sparse_matrix.py Outdated Show resolved Hide resolved
misc/sparse_matrix.py Outdated Show resolved Hide resolved
misc/sparse_matrix.py Outdated Show resolved Hide resolved
taichi/program/sparse_matrix.cpp Outdated Show resolved Hide resolved
taichi/program/sparse_matrix.h Outdated Show resolved Hide resolved
taichi/python/export_lang.cpp Outdated Show resolved Hide resolved
taichi/python/export_lang.cpp Outdated Show resolved Hide resolved
taichi/program/sparse_matrix.cpp Outdated Show resolved Hide resolved
FantasyVR and others added 2 commits August 28, 2021 12:18
Co-authored-by: Yuanming Hu <yuanming-hu@users.noreply.github.com>
@FantasyVR
Copy link
Collaborator Author

/format

Copy link
Member

@yuanming-hu yuanming-hu left a comment

Choose a reason for hiding this comment

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

LGTM now! Thank you so much for the hard work.

@yuanming-hu
Copy link
Member

I'm merging this as admin since the CI fails only because of one flaky test case on OS X. Maybe @ljcc0930 can help disable that?

@yuanming-hu yuanming-hu changed the title [Lang] Support sparse matrix [Lang] Experimental sparse matrix support on CPUs Aug 28, 2021
@yuanming-hu yuanming-hu merged commit fa45dbb into taichi-dev:master Aug 28, 2021
@FantasyVR
Copy link
Collaborator Author

LGTM now! Thank you so much for the hard work.

Sorry for cost you guys so much time to review. I learned a lot this time. Thank you very much.

@yuanming-hu
Copy link
Member

LGTM now! Thank you so much for the hard work.

Sorry for cost you guys so much time to review. I learned a lot this time. Thank you very much.

No worries. It's always challenging to merge your first essential PR at Taichi :-) People's code quality usually dramatically improves during this event. It's always nice to follow conventions and pay attention to details.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

6 participants