[Issue]: hipBLASLt support for more GPUs for PyTorch with ROCm 5.7 or later #516
Comments
|
Currently not supported on MI100. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
{{ message }}
Problem Description
PyTorch now requires hipBLASLt now when building with ROCm 5.7 or later, but hipBLASLt supports only
gfx90aGPUs.https://github.com/pytorch/pytorch/blob/84b2a323594bc7c4b47d61223b3f6466fe054416/cmake/public/LoadHIP.cmake#L158-L160
Is it means other GPUs (e.g., MI100) can not use PyTorch with the latest ROCm 6.0 release?
Operating System
Ubuntu 22.04.3 LTS
CPU
AMD EPYC 7773X
GPU
AMD Instinct MI100
Other
No response
ROCm Version
ROCm 5.7.1
ROCm Component
Pytorch
Steps to Reproduce
No response
(Optional for Linux users) Output of /opt/rocm/bin/rocminfo --support
No response
Additional Information
No response
The text was updated successfully, but these errors were encountered: