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

[FIX] Fix autoscheduler tuning on sparse matrices where there are multiple with the same shape #7974

Merged
merged 3 commits into from
May 6, 2021

Conversation

tkonolige
Copy link
Contributor

@tkonolige tkonolige commented May 4, 2021

This PR makes the autoscheduler sparse matrix registration code more specific in regards to input shapes. Instead of using density (as a two digit float) to discriminate between different inputs, we use all the input shapes. This causes fewer collisions between specifier strings. However, there are still two sparse matrices in the deploy_sparse tutorial that have the same specifier string. Ideally we would be able to use the values of the sparse matrix indices to discriminate between these two (for example, by hashing the indices), but the way registration is current setup makes this impossible. Instead we just ignore conflicts and hope that this doesn't cause a performance regression.

@jcf94

Copy link
Contributor

@jwfromm jwfromm left a comment

Choose a reason for hiding this comment

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

LGTM, thanks Tristan!

Copy link
Contributor

@jcf94 jcf94 left a comment

Choose a reason for hiding this comment

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

Thanks! It does looks better to use the specific number of data blocks.

@jcf94 jcf94 merged commit bbf7fdb into apache:main May 6, 2021
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request May 6, 2021
…tiple with the same shape (apache#7974)

* [FIX] Fix autoscheduler tuning on sparse matrices where there are multiple with the same shape

* formatting

* remove unreachable code
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request May 6, 2021
…tiple with the same shape (apache#7974)

* [FIX] Fix autoscheduler tuning on sparse matrices where there are multiple with the same shape

* formatting

* remove unreachable code
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request May 6, 2021
…tiple with the same shape (apache#7974)

* [FIX] Fix autoscheduler tuning on sparse matrices where there are multiple with the same shape

* formatting

* remove unreachable code
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request May 11, 2021
…tiple with the same shape (apache#7974)

* [FIX] Fix autoscheduler tuning on sparse matrices where there are multiple with the same shape

* formatting

* remove unreachable code
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.

3 participants