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

SparseReshape Op #7477

Merged
merged 27 commits into from
Feb 26, 2021
Merged

SparseReshape Op #7477

merged 27 commits into from
Feb 26, 2021

Conversation

codeislife99
Copy link
Contributor

@codeislife99 codeislife99 commented Feb 19, 2021

This PR adds CPU and CUDA implementation for SparseReshape Op and addresses all the issues and feedback from #7125.
Frontend Code for Tensorflow has been added. Corresponding TF Documentation
Note: This PR is a quite long , but most of it are test cases and boiler code. I can split it into CPU and CUDA implementations if you find it easier to review.

@codeislife99 codeislife99 marked this pull request as draft February 19, 2021 06:27
@codeislife99 codeislife99 mentioned this pull request Feb 20, 2021
@codeislife99 codeislife99 marked this pull request as ready for review February 20, 2021 00:25
@codeislife99
Copy link
Contributor Author

@tkonolige @mbrookhart PTAL.

python/tvm/topi/cuda/sparse_reshape.py Outdated Show resolved Hide resolved
python/tvm/topi/cuda/sparse_reshape.py Show resolved Hide resolved
python/tvm/topi/cuda/sparse_reshape.py Outdated Show resolved Hide resolved
python/tvm/topi/cuda/sparse_reshape.py Outdated Show resolved Hide resolved
tests/python/relay/test_op_level3.py Outdated Show resolved Hide resolved
Copy link
Contributor

@mbrookhart mbrookhart left a comment

Choose a reason for hiding this comment

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

Two minor comments, but overall I'm happy with where this is. @tkonolige anything else you'd like to see?

tests/python/relay/test_op_level3.py Show resolved Hide resolved
python/tvm/topi/sparse_reshape.py Show resolved Hide resolved
@codeislife99
Copy link
Contributor Author

@tkonolige All your comments have been resolved. Any other changes you'd like me to make ?

Copy link
Contributor

@tkonolige tkonolige left a comment

Choose a reason for hiding this comment

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

Looks good. Thanks @codeislife99!

@comaniac comaniac merged commit 56ac892 into apache:main Feb 26, 2021
@comaniac
Copy link
Contributor

Thanks @codeislife99 @mbrookhart @tkonolige

Lokiiiiii pushed a commit to Lokiiiiii/tvm that referenced this pull request Mar 2, 2021
* SparseReshape Inital Code

* Done

* Format

* Add empty tests

* Formatting

* SanityCheck

* formatting documentation

* Documentation

* Only Enable CPU

* Add support for CUDA

* Stuff

* Add Dynamic Support

* Parallelize GPU Impl

* Documentation

* Documentation

* Import

* Import

* Remove unnecessary code

* PR Comments

* Schedules

* Tests

* Dtypes

* Black

* Parallelize CPU

* CI error

Co-authored-by: Ubuntu <[email protected]>
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Mar 2, 2021
* SparseReshape Inital Code

* Done

* Format

* Add empty tests

* Formatting

* SanityCheck

* formatting documentation

* Documentation

* Only Enable CPU

* Add support for CUDA

* Stuff

* Add Dynamic Support

* Parallelize GPU Impl

* Documentation

* Documentation

* Import

* Import

* Remove unnecessary code

* PR Comments

* Schedules

* Tests

* Dtypes

* Black

* Parallelize CPU

* CI error

Co-authored-by: Ubuntu <[email protected]>
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.

4 participants