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

Update sparse semi-structured linear operator #104608

Closed

Conversation

alexsamardzic
Copy link
Collaborator

@alexsamardzic alexsamardzic commented Jul 4, 2023

@pytorch-bot
Copy link

pytorch-bot bot commented Jul 4, 2023

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/104608

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit 58206bc:
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@pytorch-bot pytorch-bot bot added the release notes: sparse release notes category label Jul 4, 2023
alexsamardzic added a commit that referenced this pull request Jul 4, 2023
ghstack-source-id: 379592d0bd623e5406e3b323b76ad1053103c7cd
Pull Request resolved: #104608
@alexsamardzic alexsamardzic added module: sparse Related to torch.sparse topic: not user facing topic category open source and removed release notes: sparse release notes category labels Jul 5, 2023
cc nikitaved pearu cpuhrsch amjames bhosmer

[ghstack-poisoned]
@alexsamardzic
Copy link
Collaborator Author

Just removed the code doing dense->(sparse,meta) conversion, thus sparse semi-structured linear operator simplified (renamed, for consistency, too).

alexsamardzic added a commit that referenced this pull request Jul 5, 2023
ghstack-source-id: 8129f2aaefe68e8b8d20f90c2a7df87e187c4454
Pull Request resolved: #104608
cc nikitaved pearu cpuhrsch amjames bhosmer

[ghstack-poisoned]
alexsamardzic added a commit that referenced this pull request Jul 5, 2023
ghstack-source-id: 0f87cf5922e7c4b1fe733f50cbd627d072f0ea11
Pull Request resolved: #104608
cc nikitaved pearu cpuhrsch amjames bhosmer

[ghstack-poisoned]
alexsamardzic added a commit that referenced this pull request Jul 6, 2023
ghstack-source-id: 64542b7286697baff825977c941db833ab9219f9
Pull Request resolved: #104608
cc nikitaved pearu cpuhrsch amjames bhosmer

[ghstack-poisoned]
alexsamardzic added a commit that referenced this pull request Jul 6, 2023
ghstack-source-id: 21c50a18fea6990437f99a4f0b2aba1cb7532a80
Pull Request resolved: #104608
alexsamardzic added a commit that referenced this pull request Jul 6, 2023
ghstack-source-id: 21c50a18fea6990437f99a4f0b2aba1cb7532a80
Pull Request resolved: #104608
cc nikitaved pearu cpuhrsch amjames bhosmer

[ghstack-poisoned]
alexsamardzic added a commit that referenced this pull request Jul 6, 2023
ghstack-source-id: ce61dc5e4f111301db1307528d8e4488afd80112
Pull Request resolved: #104608
alexsamardzic added a commit that referenced this pull request Jul 10, 2023
ghstack-source-id: ce61dc5e4f111301db1307528d8e4488afd80112
Pull Request resolved: #104608
cc nikitaved pearu cpuhrsch amjames bhosmer

[ghstack-poisoned]
alexsamardzic added a commit that referenced this pull request Jul 10, 2023
ghstack-source-id: f5f44ede724d761a608a736e84934ae64467e683
Pull Request resolved: #104608
alexsamardzic added a commit that referenced this pull request Jul 11, 2023
ghstack-source-id: f5f44ede724d761a608a736e84934ae64467e683
Pull Request resolved: #104608
cc nikitaved pearu cpuhrsch amjames bhosmer

[ghstack-poisoned]
alexsamardzic added a commit that referenced this pull request Jul 11, 2023
ghstack-source-id: f6305175b1584807988dc4fd3ac9b60d0dd8cabb
Pull Request resolved: #104608
cc nikitaved pearu cpuhrsch amjames bhosmer

[ghstack-poisoned]
alexsamardzic added a commit that referenced this pull request Jul 11, 2023
ghstack-source-id: cff55ce35a9967f05ef0229d876db0ab364bd1ab
Pull Request resolved: #104608
cc nikitaved pearu cpuhrsch amjames bhosmer

[ghstack-poisoned]
@@ -300,7 +300,7 @@ To use these ops, simply pass the output of ``to_sparse_semi_structured(tensor)`
>>> torch.allclose(c, torch.mm(a_sparse, b))
True

Under the hood, SparseSemiStructuredTensor will call ``torch._structured_sparse_linear`` for accelerated inference using CUTLASS sparse kernels.
Copy link
Contributor

Choose a reason for hiding this comment

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

We can remove this entirely. We don't document private functions.

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

Fixed.

Copy link
Contributor

@cpuhrsch cpuhrsch left a comment

Choose a reason for hiding this comment

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

See small comments, otherwise this is fine.

alexsamardzic added a commit that referenced this pull request Jul 13, 2023
ghstack-source-id: 0f3a2b6620a2fcd5652828e48a5e086bfeb7b519
Pull Request resolved: #104608
cc nikitaved pearu cpuhrsch amjames bhosmer

[ghstack-poisoned]
alexsamardzic added a commit that referenced this pull request Jul 13, 2023
ghstack-source-id: c3eb18fd8c02210f308fc3bd948bb74ce6ba79b1
Pull Request resolved: #104608
@alexsamardzic
Copy link
Collaborator Author

@pytorchbot merge

@pytorch-bot pytorch-bot bot added the ciflow/trunk Trigger trunk jobs on your pull request label Jul 13, 2023
@pytorchmergebot
Copy link
Collaborator

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging
Check the merge workflow status
here

@facebook-github-bot facebook-github-bot deleted the gh/alexsamardzic/16/head branch July 17, 2023 14:17
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ciflow/trunk Trigger trunk jobs on your pull request Merged module: sparse Related to torch.sparse open source topic: not user facing topic category
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants