-
Notifications
You must be signed in to change notification settings - Fork 22.1k
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
Enable Bert with Semi Structure Sparsity on ROCm #133934
base: main
Are you sure you want to change the base?
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/133934
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit ac130cb with merge base 9b76449 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
6f398db
to
892b7a1
Compare
@pytorchbot rebase |
@pytorchbot started a rebase job onto refs/remotes/origin/viable/strict. Check the current status here |
Rebase failed due to Command
Raised by https://github.com/pytorch/pytorch/actions/runs/10781630866 |
…_to_hip_mappings.py
…_to_hip_mappings.py
…_to_hip_mappings.py
…_to_hip_mappings.py
update support check for hipSparseLt cusparseLtDenseDescriptorInit update for hipsparselt
add reorder_meta function for CK wrapper add check for ROCm platform chore: Remove temporary function used for testing add TODO and fix typo
92c31ac
to
ac130cb
Compare
Draft PR. Goal is to Enable Bert with Semi Structure Sparsity on ROCm.
Refer to https://pytorch.org/tutorials/advanced/semi_structured_sparse.html
require gfx942+ and rocm 6.4
Note that this PR is built on top of #124320
cc @jeffdaily @sunway513 @jithunnair-amd @pruthvistony @ROCmSupport @dllehr-amd @jataylo @hongxiayang