-
Notifications
You must be signed in to change notification settings - Fork 2.3k
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] Support basic sparse matrix operations on GPU. #6082
Conversation
✅ Deploy Preview for docsite-preview ready!
To edit notification comments on pull requests, go to your Netlify site settings. |
By far, I have implement |
Thanks @Hanke98, I also didn't find any Cusparse API for computing element-wise multiplication. Maybe we could let it be right now and figure it out later. |
OK, thanks! Then let me add some tests and please have a reviewer later :-) |
377614b
to
702be63
Compare
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
6909c45
to
5ac0a52
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks! Could you check the failed CI test?
for more information, see https://pre-commit.ci
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM! Just a few nits.
Co-authored-by: Yi Xu <xy_xuyi@foxmail.com>
for more information, see https://pre-commit.ci
Issue: taichi-dev#2906 taichi-dev#6082 + Support element access for sparse matrix on GPU. + Add tests for GPU sparse matrix operators. Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Yi Xu <xy_xuyi@foxmail.com>
Issue: #2906
Implement the basic operations like
+
,-
,*
,@
, and transpose, etc. in this pr.