Skip to content

[ROCm] use hipSolver instead of MAGMA for Cholesky#163977

Closed
afanfa wants to merge 1 commit intopytorch:mainfrom
afanfa:improve-linalg-hip
Closed

[ROCm] use hipSolver instead of MAGMA for Cholesky#163977
afanfa wants to merge 1 commit intopytorch:mainfrom
afanfa:improve-linalg-hip

Conversation

@afanfa
Copy link
Contributor

@afanfa afanfa commented Sep 26, 2025

Currently, the Cholesky factorization and least squares operation defaults to magma when Pytorch is compiled for ROCm. This shows suboptimal performance.
This change allows PyTorch to rely on hipSolver instead of Magma.
@jeffdaily

cc @jeffdaily @sunway513 @jithunnair-amd @pruthvistony @ROCmSupport @dllehr-amd @jataylo @hongxiayang @naromero77amd

@pytorch-bot
Copy link

pytorch-bot bot commented Sep 26, 2025

🔗 Helpful Links

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

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

✅ No Failures

As of commit c863c29 with merge base 9534c59 (image):
💚 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 module: rocm AMD GPU support for Pytorch label Sep 26, 2025
@linux-foundation-easycla
Copy link

linux-foundation-easycla bot commented Sep 26, 2025

CLA Signed

  • ✅login: afanfa / name: Alessandro Fanfarillo / (c863c29)

The committers listed above are authorized under a signed CLA.

@jeffdaily jeffdaily changed the title Use hipSolver when ROCm is available instead of MAGMA for Cholesky [ROCm] use hipSolver instead of MAGMA for Cholesky Sep 26, 2025
@jeffdaily jeffdaily added release notes: rocm mandatorylabel ciflow/rocm Trigger "default" config CI on ROCm ciflow/rocm-mi300 Trigger "default" config CI on ROCm MI300 labels Sep 26, 2025
@pytorch-bot

This comment was marked as outdated.

@pytorch-bot

This comment was marked as outdated.

@pytorch-bot pytorch-bot bot removed ciflow/rocm Trigger "default" config CI on ROCm ciflow/rocm-mi300 Trigger "default" config CI on ROCm MI300 labels Sep 26, 2025
@jeffdaily jeffdaily added ciflow/rocm Trigger "default" config CI on ROCm ciflow/rocm-mi300 Trigger "default" config CI on ROCm MI300 labels Sep 26, 2025
@cyyever
Copy link
Collaborator

cyyever commented Sep 28, 2025

@pytorchmergebot merge

@pytorch-bot pytorch-bot bot added the ciflow/trunk Trigger trunk jobs on your pull request label Sep 28, 2025
@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

jainapurva pushed a commit that referenced this pull request Sep 29, 2025
Currently, the Cholesky factorization and least squares operation defaults to magma when Pytorch is compiled for ROCm. This shows suboptimal performance.
This change allows PyTorch to rely on hipSolver instead of Magma.
@jeffdaily

Pull Request resolved: #163977
Approved by: https://github.com/Skylion007
maggiemoss pushed a commit to maggiemoss/pytorch that referenced this pull request Sep 29, 2025
Currently, the Cholesky factorization and least squares operation defaults to magma when Pytorch is compiled for ROCm. This shows suboptimal performance.
This change allows PyTorch to rely on hipSolver instead of Magma.
@jeffdaily

Pull Request resolved: pytorch#163977
Approved by: https://github.com/Skylion007
jithunnair-amd pushed a commit to ROCm/pytorch that referenced this pull request Feb 2, 2026
Currently, the Cholesky factorization and least squares operation defaults to magma when Pytorch is compiled for ROCm. This shows suboptimal performance.
This change allows PyTorch to rely on hipSolver instead of Magma.
@jeffdaily

Pull Request resolved: pytorch#163977
Approved by: https://github.com/Skylion007

(cherry picked from commit 238dc65)
jithunnair-amd pushed a commit to ROCm/pytorch that referenced this pull request Feb 2, 2026
Currently, the Cholesky factorization and least squares operation defaults to magma when Pytorch is compiled for ROCm. This shows suboptimal performance.
This change allows PyTorch to rely on hipSolver instead of Magma.
@jeffdaily

Pull Request resolved: pytorch#163977
Approved by: https://github.com/Skylion007

(cherry picked from commit 238dc65)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/rocm Trigger "default" config CI on ROCm ciflow/rocm-mi300 Trigger "default" config CI on ROCm MI300 ciflow/trunk Trigger trunk jobs on your pull request Merged module: rocm AMD GPU support for Pytorch open source release notes: rocm mandatorylabel

Projects

None yet

Development

Successfully merging this pull request may close these issues.

6 participants